加载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)
相关推荐
外收内放7 小时前
数据分析(数据读取与写入)
python·学习·数据分析
zh路西法7 小时前
【离策略强化学习】DDPG、TD3 与 SAC:核心公式推导与源码解析
python·强化学习·ddpg·td3·sb3·off-policy·sac
秋秋小事7 小时前
python 数据契约层FastAPI+Pydantic(一)
python
AIFQuant7 小时前
Python港股实时行情API接入:恒生股票数据获取与AH股溢价监控实战
开发语言·数据库·python·金融·实时数据api
FLJwu7 小时前
硬件器件选型实战电源篇 03:TVS 管选型踩坑复盘|盲目选大电压,静电打坏运动相机主控,客户批量退货
经验分享·数码相机·音视频·相机·运动相机
朝阳资本论7 小时前
小马智行杀入伦敦牵手Uber,中国Robotaxi出海怎么打?
python
外收内放7 小时前
Python基础语法练习题(70)
开发语言·python
老唐7777 小时前
Agent 开发入门教程:从零搭建一个 Python AI Agent
人工智能·python·深度学习·langchain·大模型·agent·智能体
宿州派大星10 小时前
Python实战:PyQt5 + OpenCV + Dlib 打造AI桌面应用
人工智能·python·qt
承渊政道11 小时前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(开源大模型ChatGLM高级定制化应用实战)
人工智能·pytorch·python·llm·模型微调·glm