加载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)
相关推荐
少云清11 分钟前
【金融项目实战】7_接口测试 _代码实现接口测试(重点)
python·金融项目实战
深蓝电商API12 分钟前
爬虫IP封禁后的自动切换与检测机制
爬虫·python
m0_5500246313 分钟前
持续集成/持续部署(CI/CD) for Python
jvm·数据库·python
B站_计算机毕业设计之家1 小时前
豆瓣电影数据采集分析推荐系统 | Python Vue Flask框架 LSTM Echarts多技术融合开发 毕业设计源码 计算机
vue.js·python·机器学习·flask·echarts·lstm·推荐算法
渣渣苏1 小时前
Langchain实战快速入门
人工智能·python·langchain
王锋(oxwangfeng)1 小时前
YOLOWorld 实现开集障碍物检测
yolo
lili-felicity1 小时前
CANN模型量化详解:从FP32到INT8的精度与性能平衡
人工智能·python
数据知道1 小时前
PostgreSQL实战:详解如何用Python优雅地从PG中存取处理JSON
python·postgresql·json
喵叔哟1 小时前
02-YOLO-v8-v9-v10工程差异对比
人工智能·yolo·机器学习
ZH15455891311 小时前
Flutter for OpenHarmony Python学习助手实战:面向对象编程实战的实现
python·学习·flutter