您现在的位置是：首页 >学无止境 >Python —— Windows10下训练Yolov5分割模型并测试网站首页学无止境

Python —— Windows10下训练Yolov5分割模型并测试

信必诺 2023-07-13 15:06:18
简介Python —— Windows10下训练Yolov5分割模型并测试

 
附：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虚拟

风语者！平时喜欢研究各种技术，目前在从事后端开发工作，热爱生活、热爱工作。

站长推荐