附:Python —— Windows10下配置Pytorch环境、进行训练模型并测试(完整流程,附有视频)
效果
手机拍摄一段工位视频,上传到win10训练了yolov5分割鼠标的样本后推理效果截图。
训练准备
1、查看自己下载的Yolov5源码是否存在"segment"文件夹,该文件夹下存在分割的训练与推理、测试的py代码。若不存在更新更高版本v5源码。
2、下载 yolov5s-seg.pt (提取码:lpqa),存放到源码的weights文件夹内。之后可以测试一下:
python segment/predict.py --weights segment/yolov5s-seg.pt --source 4.mp4
3、安装labelme工具
由于我win10电脑安装conda,于是在conda虚拟