加载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)
相关推荐
沈浩(种子思维作者)15 小时前
真的能精准医疗吗?癌症能提前发现吗?
人工智能·python·网络安全·健康医疗·量子计算
ASF1231415sd15 小时前
【基于YOLOv10n-CSP-PTB的大豆花朵检测与识别系统详解】
人工智能·yolo·目标跟踪
njsgcs16 小时前
ue python二次开发启动教程+ 导入fbx到指定文件夹
开发语言·python·unreal engine·ue
io_T_T16 小时前
迭代器 iteration、iter 与 多线程 concurrent 交叉实践(详细)
python
华研前沿标杆游学16 小时前
2026年走进洛阳格力工厂参观游学
python
Carl_奕然16 小时前
【数据挖掘】数据挖掘必会技能之:A/B测试
人工智能·python·数据挖掘·数据分析
AI小怪兽17 小时前
基于YOLOv13的汽车零件分割系统(Python源码+数据集+Pyside6界面)
开发语言·python·yolo·无人机
wszy180917 小时前
新文章标签:让用户一眼发现最新内容
java·python·harmonyos
Eric.Lee202117 小时前
python实现 mp4转gif文件
开发语言·python·手势识别·手势交互·手势建模·xr混合现实
EntyIU17 小时前
python开发中虚拟环境配置
开发语言·python