欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

计算视频FPS

程序员文章站 2022-05-29 10:02:18
'''功能:计算算法预测视频时的速度FPS,即1秒检测多少帧'''import cv2import numpy as npFPSES=[]import timedef 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,即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