1 Star 0 Fork 254

YELLOWZAN / PaddleDetection

forked from PaddlePaddle / PaddleDetection 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
README.md 4.20 KB
一键复制 编辑 原始数据 按行查看 历史

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}
}
Python
1
https://gitee.com/yellowzan/PaddleDetection.git
git@gitee.com:yellowzan/PaddleDetection.git
yellowzan
PaddleDetection
PaddleDetection
release/2.5

搜索帮助