关于之前pytorch框架下yolov3推理和训练环境搭建可以参考,本文所使用的环境和这篇记录一致。
pytorch yolov3 推理和训练环境搭建_papaofdoudou的博客-CSDN博客_yolov3环境搭建
下载代码框架环境:
git clone https://github.com/ultralytics/yolov5
安装依赖:
pip install -r requirements.txt
下载YOLOV5模型文件:
下载链接 :Releases · ultralytics/yolov5 · GitHub以下模型均可用:
拷贝模型文件到yolov5顶层目录,之后执行命令:
python detect.py --source data/images --weights yolov5s.pt --conf 0.25
输出如下,一开始会去下载字库,猜测是用来画标签用:
根据LOG,推理结果在runs/detect/exp目录:
yolov5s6.pt模型推理
python detect.py --source data/images --weights yolov5s6.pt --conf 0.25
排错:
如果遇到下面的错误,说明PROTOBUF版本不对,需要重新安装。
安装命令:
pip install protobuf==3.20.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
之后就可以了。
python代码推理:
由于我们已经将YOLOV5的pytorch训练框架下载过来,接下来我们就可