上代码
import time
import datetime
class Timer(object):
def __init__(self):
self.init_time = time.time()
self.total_time = 0.
self.calls = 0
self.start_time = 0.
self.diff = 0.
self.average_time = 0.
self.remain_time = 0.
def tic(self):#实现时间初始化
self.start_time = time.time()
def toc(self, average = True):#计算平均时间
self.diff = time.time() - self.start_time
self.total_time += self.diff
self.call +=1
self.average_time = self.total_time / self.calls
if average:
return self.average_time
else:
return self.diff
def remain(self,iters,max_iters): # 剩余多长时间
if iters == 0:
self.remain_time = 0
else:
self.remain_time = (time.time() - self.init_time) / iters \ * (max_iters - iters)
return str(datetime.timedelta(seconds = int(self.remain_time))) #转化为时分秒的格式
t = Timer()
t.tic()
time.sleep(3)
print(t.toc()) #3.00016450881958
a = t.remain(1,3)
print("remain time:%s"%a) #remain time:0:00:06
具体例子见代码注释。
下一篇见详细介绍yolo算法读取数据集及标签的代码