- 项目应用场景
-
- 面向钢管焊缝缺陷检测场景,使用深度学习算法来实现,提供钢管焊缝缺陷检测数据集,数据集已经标注整理好,包括 YOLO 和 PASCAL VOC 数据格式,项目检出效果好。
- 训练数据集展示
- 项目效果
- 项目细节 ==> 具体参见项目
README.md
-
- (1) 安装依赖
bash
pip install -r requirements.txt
-
- (2) 下载训练数据集,数据集下载地址在 README
bash
# unzip datasets
unzip steel-tube-dataset-all.zip
-
- (3) 执行训练
bash
python ./yolov5/train.py
- 项目获取