一、网络结构
二、输入输出
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 detect.py --source ./datasets/tooth_DA/test/images --weights ./runs/train/exp2/weights/best.pt
测试完成后得到了图像 .../yolo5/runs/detect/exp6