我要评论
'''
功能:计算算法预测视频时的速度FPS,即1秒检测多少帧
'''
import cv2
import numpy as np
FPSES=[]
import time
def predict_method():#模拟算法用时
time.sleep(0.01)
def main():
video_path= './test_video/001.mp4'
cap = cv2.VideoCapture(video_path)
cap.set(cv2.CAP_PROP_FPS, 30)
while cap.isOpened():
ret, image = cap.read()
prev_time = time.time()
predict_method()
fps_my = 1 / (time.time() - prev_time)
if len(FPSES) < 30:
FPSES.append(fps_my)
else:
del FPSES[0]
FPSES.append(fps_my)
FPS_yolo = int(np.mean(FPSES)) # 我这里取均值
cv2.putText(image, "FPS: %f" % (FPS_yolo), (int(20), int(40)), 0, 5e-3 * 200, (0, 255, 0), 3)
if not ret:
break
cv2.imshow("car detect", image)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cv2.destroyAllWindows()
if __name__ == '__main__':
main()
本文地址：https://blog.csdn.net/weixin_38145317/article/details/107357919
您可能感兴趣的文章:
如您对本文有疑问或者有任何想说的，请点击进行留言回复，万千网友为您解惑！
网友评论