yolov9实现图像分割(gelan-c-seg.pt)步骤

人生忽如寄-

已于 2024-09-11 10:41:41 修改

阅读量356

点赞数 7

文章标签： YOLO python 目标检测计算机视觉

于 2024-08-29 09:59:21 首次发布

本文链接：https://blog.csdn.net/weixin_48679164/article/details/141666982

版权

创建虚拟环境

conda activate yolov9 -n python=3.8
conda acitivate yolov9
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

数据准备

在主文件夹创建一个data.yaml文件
data(images\labels)
splitdata.py(train\test\val)

训练

models/segment/gelan-c-seg.yaml这个文件里面修改nc值为1

python segment/train.py --workers 8 --device 0 --batch 32  --data data.yaml --img 640 --cfg models/segment/gelan-c-seg.yaml --weights 'gelan-c-seg.pt' --name gelan-c-seg --hyp hyp.scratch-high.yaml --no-overlap --epochs 100 --close-mosaic 10

如果意外中断训练，可从断点继续训练

python segment/train.py --workers 8 --device 0 --batch 32 --data data.yaml --img 640 --cfg models/segment/gelan-c-seg.yaml --weights 'gelan-c-seg.pt' --name gelan-c-seg --hyp hyp.scratch-high.yaml --no-overlap --epochs 100 --close-mosaic 10 --resume 'C:/.../yolov9-main/yolov9-main/runs/train-seg/gelan-c-seg7/weights/last.pth'

验证

修改segment/val.py中的run()函数：save_json=True/save_txt=True…

python segment/val.py --workers 8 --device 0 --batch 32 --data data.yaml --img 640 --weights "C:/Users/.../runs/train-seg/gelan-c-seg7/weights/best.pt" --name gelan-c-seg-val --save-json --iou-thres 0.6 --conf-thres 0.001