PP-YOLOE+ 下游任务
我们验证了PP-YOLOE+模型强大的泛化能力,在农业、低光、工业等不同场景下游任务检测效果稳定提升!
农业数据集采用Embrapa WGISD,该数据集用于葡萄栽培中基于图像的监测和现场机器人技术,提供了来自5种不同葡萄品种的实地实例,
处理后的COCO格式,包含图片训练集242张,测试集58张,5个类别,Embrapa WGISD COCO格式下载;
低光数据集使用ExDark,该数据集是一个专门在低光照环境下拍摄出针对低光目标检测的数据集,包括从极低光环境到暮光环境等10种不同光照条件下的图片,
处理后的COCO格式,包含图片训练集5891张,测试集1472张,12个类别,ExDark COCO格式下载;
工业数据集使用PKU-Market-PCB,该数据集用于印刷电路板(PCB)的瑕疵检测,提供了6种常见的PCB缺陷,
处理后的COCO格式,包含图片训练集555张,测试集138张,6个类别,PKU-Market-PCB COCO格式下载。
实验结果:
模型 |
数据集 |
mAPval 0.5:0.95 |
下载链接 |
配置文件 |
PP-YOLOE_m |
Embrapa WGISD |
52.7 |
下载链接 |
配置文件 |
PP-YOLOE+_m (obj365_pretrained) |
Embrapa WGISD |
60.8(+8.1) |
下载链接 |
配置文件 |
PP-YOLOE+_m (coco_pretrained) |
Embrapa WGISD |
59.7(+7.0) |
下载链接 |
配置文件 |
PP-YOLOE_m |
ExDark |
56.4 |
下载链接 |
配置文件 |
PP-YOLOE+_m (obj365_pretrained) |
ExDark |
57.7(+1.3) |
下载链接 |
配置文件 |
PP-YOLOE+_m (coco_pretrained) |
ExDark |
58.1(+1.7) |
下载链接 |
配置文件 |
PP-YOLOE_m |
PKU-Market-PCB |
50.8 |
下载链接 |
配置文件 |
PP-YOLOE+_m (obj365_pretrained) |
PKU-Market-PCB |
52.7(+1.9) |
下载链接 |
配置文件 |
PP-YOLOE+_m (coco_pretrained) |
PKU-Market-PCB |
52.4(+1.6) |
下载链接 |
配置文件 |
注意:
- PP-YOLOE模型训练过程中使用8 GPUs进行训练,如果GPU卡数或者batch size发生了改变,你需要按照公式 lrnew = lrdefault * (batch_sizenew * GPU_numbernew) / (batch_sizedefault * GPU_numberdefault) 调整学习率。
- 具体使用教程请参考ppyoloe。
引用
@article{Exdark,
title={Getting to Know Low-light Images with The Exclusively Dark Dataset},
author={Loh, Yuen Peng and Chan, Chee Seng},
journal={Computer Vision and Image Understanding},
volume={178},
pages={30-42},
year={2019},
doi={https://doi.org/10.1016/j.cviu.2018.10.010}
}