Opencv库调用yolov5模型onnx格式，识别视频，反馈进度，保存结果

最新推荐文章于 2025-05-15 20:17:56 发布

原创

最新推荐文章于 2025-05-15 20:17:56 发布 · 913 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#opencv #YOLO #音视频

上！代！码！

import cv2
import numpy as np
import onnxruntime as rt

# Load the ONNX model
sess = rt.InferenceSession("yolov5s.onnx")

# Load the video
cap = cv2.VideoCapture('A1.mp4')


# Get video info
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))  # Get total number of frames

# Define the codec and create a VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
out = cv2.VideoWriter('A2.mp4', fourcc, fps, (width, height))

frame_count = 0  # Initialize frame counter

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        # Preprocess the frame
        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        frame_resized = cv2.resize(frame_rgb, (640, 640))
        frame_normalized = frame_resized.astype(np.float32) / 255.0
        frame_reshaped = np.transpose(frame_norma