Python —— Windows10下训练Yolov5分割模型并测试_yolov5s-seg.pt

 
附：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虚拟环境下安装labelme工具供之后扣样本使用。

(注意:大家可直接搞第三步。)

1、打开"Anaconda Prompt (install)"命令行

2、切换到pytorch1.8.1
	conda activate pytorch1.8.1

3、下载安装labelme
	// (若该虚拟环境没安装pyqt则进行安装,作者在装conda时已装过所以没有输入该条指令。)
	pip install pyqt5 -i https://pypi.tuna.tsinghua.edu.cn/simple	

	// 下载安装labelme（以下两条指令均可）
	pip install labelme
	pip install labelme -i https://pypi.tuna.tsinghua.edu.cn/simple
1
2
3
4
5
6
7
8
9
10
11
12
13
14

 
     4、使用labelme工具抠出鼠标并保存json文件

 
     5、将labelme保存的json文件统统转为txt，供训练使用

import json
import os
1
2