加载YOLO模型,处理mp4视频

已经有训练好的YOLO模型,执行命令如下,其中show是选择是否一遍处理一遍预览处理后的视频。
模型和mp4视频资源

cpp 复制代码
yolo predict model=best.pt source=car.mp4  project=. name=output_video show=True

以下是代码的方式完成,会把画好框,带置信度的视频保存下来。

cpp 复制代码
from ultralytics import YOLO
import cv2

# 你的模型路径(best.pt)
model = YOLO("best.pt")

# 输入视频
input_path = "car.mp4"
output_path = "output.mp4"

cap = cv2.VideoCapture(input_path)

# 获取视频信息
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
fps = cap.get(cv2.CAP_PROP_FPS)
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # YOLO 推理
    results = model(frame, verbose=False)

    # 绘制检测结果
    annotated_frame = results[0].plot()  # 自动画框/标签/置信度

    # 写入输出视频
    out.write(annotated_frame)

cap.release()
out.release()
print("✅ 保存完成:", output_path)
相关推荐
明月醉窗台9 分钟前
深度学习(17)YOLO训练中的超参数详解
人工智能·深度学习·yolo
weixin1997010801614 分钟前
[特殊字符] 智能数据采集:数字化转型的“数据石油勘探队”(附Python实战源码)
开发语言·python
次元工程师!1 小时前
LangFlow开发(三)—Bundles组件架构设计(3W+字详细讲解)
java·前端·python·低代码·langflow
t_hj1 小时前
大模型微调
人工智能·python·深度学习
范范@2 小时前
python基础-函数
开发语言·python
2301_803934612 小时前
MySQL 字段类型选择规范指南
jvm·数据库·python
yaoxin5211233 小时前
406. Java 文件操作基础 - 字符与二进制流
java·开发语言·python
wj3055853783 小时前
课程 6:图生视频首次运行流程
人工智能·音视频
一勺菠萝丶4 小时前
macOS 安装 Python 包报错:`externally-managed-environment` 怎么解决?
python
runafterhit4 小时前
显示调研专题-OLED 终端市场分析报告
音视频