yolov5流程

1.将接受到的信号图像使用labelImg软件进行打标,注意打标过后生成的xml文件数量要和图片数量保持一致。如果两者数量不一致,可执行check_file.py程序a3c26f69464c441e9599c62981c8e3a2.jpg

 labpath为xml文件所在路径。

2.执行xml2txt.py程序,将xml转换为TXT文件,具体代码网上有。除修改路径外,注意classes即标签的选择,与打标时选择的标签数有关。

3.将图片放在数据集images下的train或val里,将TXT文件放在labels下的train或val里(与图片存放位置对应)。

4.修改(或新创)yolov5/data下的yaml文件中数据集的路径6d9f7020cdb94fbface7091a57e2df41.jpg

 5.修改train.py中的一些参数,参考网上。执行python train.py --data data0217.yaml --cfg yolov5m.yaml --weight '' --batch -size 64 --imgsz 512 --device 0,1,2,3开始训练。若训练图像为矩形如1024*512 则在末尾加--rect

使用labelme软件格式为json,流程大致相同,格式转换时使用json2txt.py,具体代码网上有。

6.python detect.py --weight ./runs/train/exp33/weights/best.pt --source  ./datasets/images/val  --imgsz 512 -- save-conf 检测

python val.py --data test_dect.yaml --weight  ./runs/train/exp33/weights/best.pt --batch-size 64 --iou-thres 0.65  -task test --save-txt --save-conf

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值