yolov5训练识别钢筋模型

使用yolov5训练识别钢筋的模型,用于yolov5识别钢筋并计数(yolov5钢筋计数)

 

 

1、标注数据

 

2、整理数据

3、训练:修改:myvoc.yaml

myvoc.yaml
train: VOC_2022042401/train.txt
val: VOC_2022042401/val.txt

# number of classes
nc: 1

# class names
names: ["rebar"]

 

4、开始训练

python train_20220424.py --img-size 640 --batch-size 1 --epochs 300 --data ./data/myvoc.yaml --cfg ./models/yolov5m.yaml --workers 0

 

训练日志:

  

5、训练完毕,查看模型相关参数

 

6、调用最优模型测试

python detect_imgs_222042401.py --weights runs/train/exp10/weights/best.pt --source data/img

 

测试效果:

 

 

 

 

 

 

############################

posted @   西北逍遥  阅读(378)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~
历史上的今天:
2021-04-24 计算mAP
2021-04-24 mask_rcnn调用训练好的模型
2021-04-24 class list
2020-04-24 python opencv Mat pyqt5 QPixmap
点击右上角即可分享
微信分享提示