【笔记】通过UltralyticsYOLO实现视频识别

发表于 更新于 阅读次数：

前言

通过UltralyticsYOLO实现视频识别

下载依赖

下载PyTorch

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

下载Ultralytics

1
pip3 install ultralytics

下载模型

1
wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8s.pt

通过命令行工具实现视频识别

物体检测

model=<file>.pt：指定模型文件
source=<video>.mp4：指定输入的视频文件

1
yolo predict model=<file>.pt source=<video>.mp4 save_txt=True save_conf=True save_crop=True
  • 结果会生成到runs/detect目录下

完成

参考文献

Ultralytics官方文档