YOLOv5s处理二维牙齿数据集

一、网络结构

二、输入输出

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

得出了权重 .../yolo