Python Timer和TimerFPS计时工具类
作者:小锋学长生活大爆炸
这篇文章主要介绍了Python Timer和TimerFPS计时工具类,文章分享得代码内容详细,具有一的的参考价值,需要的小伙伴可以参考一下
以下工具类代码来自开源项目pyslam
。
Timer
import cv2 class Colors(object): ''' Colors class:reset all colors with colors.reset; two sub classes fg for foreground and bg for background; use as colors.subclass.colorname. i.e. colors.fg.red or colors.bg.greenalso, the generic bold, disable, underline, reverse, strike through, and invisible work with the main class i.e. colors.bold ''' reset='\033[0m' bold='\033[01m' disable='\033[02m' underline='\033[04m' reverse='\033[07m' strikethrough='\033[09m' invisible='\033[08m' class fg: black='\033[30m' red='\033[31m' green='\033[32m' orange='\033[33m' blue='\033[34m' purple='\033[35m' cyan='\033[36m' lightgrey='\033[37m' darkgrey='\033[90m' lightred='\033[91m' lightgreen='\033[92m' yellow='\033[93m' lightblue='\033[94m' pink='\033[95m' lightcyan='\033[96m' class bg: black='\033[40m' red='\033[41m' green='\033[42m' orange='\033[43m' blue='\033[44m' purple='\033[45m' cyan='\033[46m' lightgrey='\033[47m' class Printer(object): @staticmethod def red(*args, **kwargs): print(Colors.fg.red, *args, **kwargs) print(Colors.reset, end="") @staticmethod def green(*args, **kwargs): print(Colors.fg.green, *args, **kwargs) print(Colors.reset, end="") @staticmethod def blue(*args, **kwargs): print(Colors.fg.blue, *args, **kwargs) print(Colors.reset, end="") @staticmethod def cyan(*args, **kwargs): print(Colors.fg.cyan, *args, **kwargs) print(Colors.reset, end="") @staticmethod def orange(*args, **kwargs): print(Colors.fg.orange, *args, **kwargs) print(Colors.reset, end="") @staticmethod def purple(*args, **kwargs): print(Colors.fg.purple, *args, **kwargs) print(Colors.reset, end="") @staticmethod def yellow(*args, **kwargs): print(Colors.fg.yellow, *args, **kwargs) print(Colors.reset, end="") @staticmethod def error(*args, **kwargs): print(Colors.fg.red, *args, **kwargs, file=sys.stderr) print(Colors.reset, end="") #timer_print = print timer_print = Printer.cyan class Timer: def __init__(self, name = '', is_verbose = False): self._name = name self._is_verbose = is_verbose self._is_paused = False self._start_time = None self._accumulated = 0 self._elapsed = 0 self.start() def start(self): self._accumulated = 0 self._start_time = cv2.getTickCount() def pause(self): now_time = cv2.getTickCount() self._accumulated += (now_time - self._start_time)/cv2.getTickFrequency() self._is_paused = True def resume(self): if self._is_paused: # considered only if paused self._start_time = cv2.getTickCount() self._is_paused = False def elapsed(self): if self._is_paused: self._elapsed = self._accumulated else: now = cv2.getTickCount() self._elapsed = self._accumulated + (now - self._start_time)/cv2.getTickFrequency() if self._is_verbose is True: name = self._name if self._is_paused: name += ' [paused]' message = 'Timer::' + name + ' - elapsed: ' + str(self._elapsed) timer_print(message) return self._elapsed
用法
import Timer timer = Timer(is_verbose=True) timer.start() # 此处是你的代码 timer.elapsed()
效果
TimerFps
在上面的基础上,再加上下面代码段:
import os import numpy as np import cv2 import math class MovingAverage: def __init__(self, average_width = 10, compute_sigma = False): self._average_width = average_width self._idx_ring = 0 self._average = 0 self._sigma2 = 0 self._is_init = False self._is_compute_sigma = compute_sigma self._one_over_average_width_min_one = 1./(average_width-1) self._ring_buffer = np.zeros(average_width) def init(self, initVal=None): if initVal is None: initVal = 0. self._ring_buffer = np.full(self._average_width, initVal, dtype=np.float32) self._average = initVal; self._sigma2 = 0; self._is_init = True; def getAverage(self, new_val=None): if not self._is_init: self.init(new_val) if new_val is None: return self._average averageOld = self._average oldVal = self._ring_buffer[self._idx_ring] self._average += (new_val - oldVal)/self._average_width if self._is_compute_sigma: self._sigma2 = self._sigma2 + self._one_over_average_width_min_one*(self._average_width*(averageOld*averageOld - self._average*self._average) - oldVal*oldVal + newVal*newVal) self._ring_buffer[self._idx_ring] = new_val self._idx_ring = (self._idx_ring + 1) % self._average_width return self._average def getSigma(self): return math.sqrt(max(self._sigma2,0.)) class TimerFps(Timer): def __init__(self, name='', average_width = 10, is_verbose = True): super().__init__(name, is_verbose) self.moving_average = MovingAverage(average_width) def refresh(self): elapsed = self.elapsed() self.moving_average.getAverage(elapsed) self.start() if self._is_verbose is True: dT = self.moving_average.getAverage() name = self._name if self._is_paused: name += ' [paused]' message = 'Timer::' + name + ' - fps: ' + str(1./dT) + ', T: ' + str(dT) timer_print(message)
用法
import TimerFps timer = TimerFps(name='') timer.start() # 这里是你的代码 timer.refresh()
效果
到此这篇关于Python Timer和TimerFPS计时工具类的文章就介绍到这了,更多相关Python 计时工具类内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!