YOLO v7 在不同硬體下, 判讀影片的速度
文章目錄
基於上次 在 Colab 上用 YOLOv7 判讀照片, 以下測試在不同硬體跑 YOLOv7 來判讀影片.
資料
原始影像: https://pixabay.com/videos/pigeon-animal-bird-background-5938/ (約 15 秒)
指令
1conda create -n yolov7 python=3.10 -y
2conda activate yolov7
3pip install -r requirements.txt
4
5python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source pigeon.mp4 --view-img --device cpu
跑出來的數據
其實和預期的差不多, GPU 真的快很多.
| CPU (or GPU) specification | yolov7.pt | yolov7-e6e.pt | 環境 |
|---|---|---|---|
| AMD Ryan 7 Pro 5850U (8core) | 263s | 415s | 筆電 |
| Intel Xeon Gold 6154 CPU (32core) | 115s | 216s | Server |
| Tesla V100-SXM2-32GB | 8.3s | 10.8s | 某雲 |