YOLOv5s处理二维牙齿数据集

本文详细介绍了使用YOLOv5进行图像识别的过程,包括网络结构、输入640x640图像的处理、权重文件生成,以及数据预处理、训练、验证和测试的具体步骤。作者展示了如何下载模型、安装环境、执行训练和测试命令,并分享了关键输出路径。
摘要由CSDN通过智能技术生成

一、网络结构

二、输入输出

1、输入

640x640的图像

2、输出

权重文件

测试图像

三、数据预处理

在github上下载YOLOv5的模型,并安装模型所需环境

pip install -U -r requirements.txt

四、训练&测试

对数据集进行训练

python train.py --img 640 --batch 16 --epochs 100 --data ./data/tooth_data.yaml --cfg ./models/yolov5s.yaml --weights ./yolov5s.pt

得出了权重 .../yolo5/runs/train/exp2/weights/best.pt

对训练出来的权重进行验证

python val.py --weights ./runs/train/exp19/weights/best.pt --data ./data.yaml --half --imgsz 640

得到了验证集的结果 Results saved to runs/val/exp3

开始用权重测试

python detect.py --source ./datasets/tooth_DA/test/images --weights ./runs/train/exp2/weights/best.pt

测试完成后得到了图像 .../yolo5/runs/detect/exp6

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值