极速入门!用yolov5看西游记识别小怪物!!!

1,257 阅读1分钟

准备好一张GPU的卡,另外电脑配置要好!

安装docker环境,版本最好在19以上。

下载yolov5的docker镜像,命令:

docker pull ultralytics/yolov5

然后漫长的等待开始运行容器命令,启动容器本次使用GPU

docker run --ipc=host --gpus all -it ultralytics/yolov5:latest

当前的工作目录:/usr/local/app

开始进行识别,本次使用在线视频地址:

python detect.py --source vd3.bdstatic.com/mda-km7nk7i…

结果如下:

截屏2021-06-30 下午5.04.28.png 程序员用yolo看西游记能识别妖怪码? - 千年码农的视频 www.zhihu.com/zvideo/1393…