计算视频FPS

'''
功能:计算算法预测视频时的速度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()