加载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)
相关推荐
zoujiahui_20182 分钟前
python中模型加速训练accelerate包的用法
开发语言·python
民乐团扒谱机4 分钟前
【微实验】基于Python实现的实时键盘鼠标触控板拾取检测(VS2019,附完整代码)
python·c#·计算机外设
jinglong.zha10 分钟前
【Yolov8】图形化检测视频-源码免费分享
人工智能·yolo·目标跟踪·视觉检测·yolov8·yolov11
心本无晴.12 分钟前
拣学--基于vue3和django框架实现的辅助考研系统
vue.js·python·mysql·考研·django·dify
Darenm11112 分钟前
关于AI 面试官项目:智选ai 基于 Vue3 + Django + Dify 的全栈开发实战
人工智能·python·django
大写-凌祁14 分钟前
Change3D:从视频建模视角重新审视变化检测与描述
3d·音视频
詹姆斯爱研究Java15 分钟前
基于Django的租房网站的设计与实现
数据库·python·django
拉姆哥的小屋18 分钟前
基于多模态深度学习的城市公园社交媒体评论智能分析系统——从BERTopic主题建模到CLIP图文一致性的全栈实践
人工智能·python·深度学习·矩阵·媒体
ZAz_19 分钟前
DAY 41 图像数据与显存
python
曲幽22 分钟前
Python环境管理利器Conda:从入门到避坑实战指南
python·conda·pip·anaconda·uv·venv·miniconda