1.下载准备
从https://github.com/ultralytics/yolov5中下载所对应的文件。并且提前下载好对应的预训练文件,从中任选一个Releases · ultralytics/yolov5 · GitHub
2.配置环境
使用conda虚拟环境安装所需要的包。(假设已经安装完成anaconda,安装步骤可参考相关连接)
1.创建一个新的虚拟环境:
conda activate yolo python=3.9
这里yolo表示你创建的虚拟环境的名称,3.9表示你的python版本(这里要求python的版本号>=1.8)
2.安装requirements.txt
pip install -r requirements.txt
如果安装失败了,我们可以通过pip命令进行手动的安装,因为txt文件一旦遇到某个包安装失败,就会自动停止安装,导致后面的包没有安装成功。
3.测试模型
安装成功后,进入segment目录,找到predict文件,可以看到这里有很多参数,通常,我们运行的时候加上自己的模型路径,加上图片的路径,以及运行环境就可以。例如
python detect.py --weights G:\workspaces\yolov5n-seg.pt --source G:\workspaces\yolov5-master\yolov5-master\data\images\bus.jpg
详细训练过程待添加,敬请期待。。。。