PaddleDetection v2.6发布:目标小?数据缺?标注累?泛化差?PP新员逐一应对!
从2022年10月份开始,大家已经陆续在PaddleDetection仓库的develop分支下尝鲜体验了一些预发布的新算法。如今经过4个月的打磨升级,PaddleDetection v2.6正式发布,最新版本有哪些更新以及性能提升呢?快来打开这份更新解读吧!
更新重点
PP-YOLOE家族再添新员,半监督、少样本、蒸馏学习加持,旋转框、小目标、密集全场景检测覆盖;
PP-Human、PP-Vehicle:功能新增、推理提速、多路视频流支持; 半监督检测、少样本学习、模型蒸馏算法全面开源,助力解决少样本、泛化性、标注量、冷启动问题。
一、PP-YOLOE家族再添新员
GEEK TALK
训练收敛加速
下游任务泛化性显著提升
高性能部署能力
了解更多
我们补充了 旋转框检测 、 小目标检测 、 密集检测 场景 专用模型
同时针对端侧GPU场景的超高推理速度要求,进一步补充了 超 轻量 通用目标检测模型 PP-YOLOE+_t 版本 同时我们提供了这些通用模型和专用模型(旋转框检测、小目标检测、密集检测)的半监督学习、少样本学习、模型蒸馏技术方案,助力解决业务场景中少样本、泛化性、标注量、冷启动等难题。
PP-YOLOE+旋转框、小目标、密集全场景检测覆盖
PP-YOLOE-R:旋转框检测模型
了解更多
PP-YOLOE-SOD:小目标检测模型
PP-YOLOE-SOD(Small Object Detection)针对小目标检测提出了两种检测方案,分别是基于切图拼图流程优化的小目标检测方案以及基于原图模型算法优化的小目标检测方案。
同时提供了数据集自动分析脚本,只需输入数据集标注文件,便可得到数据集统计结果,辅助判断数据集是否是小目标数据集以及是否需要采用切图策略,同时给出网络超参数参考值。
在VisDrone-DET数据集上,单模型精度达到38.5mAP。
了解更多
https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.6/configs/smalldet
PP-YOLOE-DOD:密集检测模型
PP-YOLOE-DOD(Dense Object Detection)是基于PP-YOLOE+的密集目标检测算法,针对密集检测任务,我们做了如下优化:
数据预处理
为了适应商超数据的高分辨率,我们增大了模型训练和推理的输入尺寸;
标签匹配策略
为了适应密集场景,我们去除了静态匹配策略,在全部的训练周期中都使用动态匹配策略。
在SKU数据集检测精度mAP 60.3,达到了开源最优。
了解更多
PP-YOLOE+_t:超轻量通用目标检测模型
了解更多
PP-YOLOE+少样本学习方案
PaddleDetection团队针对PP-YOLOE+提供了少样本学习的方案,引入了Compare-Contrast算法。
在预训练模型的基础上,仅通过极少的标注数据即可训练得到较好的模型效果。
下表是PP-YOLOE+_s版本模型在工业数据集PKU-Market-PCB上,通过引入少样本学习方案,Box AP得到了2.4的提升。
了解更多
https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.6/configs/few-shot
PP-YOLOE+半监督学习方案
PaddleDetection团队结合Dense Teacher前沿算法,针对PP-YOLOE+提供了半监督学习方案。
半监督学习结合有标签数据和无标签数据,在大幅节省数据标注的情况下,依然达到较高的模型精度。在实际产业应用过程中,半监督学习是项目冷启动时常见的策略之一。
下表是,在仅采用5%、10%有标签数据进行监督学习,95%、90%无标签数据进行半监督学习的情况下,精度得到了1.2~2.5的提升。
了解更多
PP-YOLOE+模型蒸馏方案
了解更多
二、PP-Human、PP-Vehicle功能新增、推理提速
GEEK TALK
PP-Human:行人分析工具箱,推理提速、多路视频流支持
在 PP-YOLOE+_t 模型的加持下,PP-Human行人分析工具箱开源 边缘端实时检测模型 ,精度mAP 45.7, Jetson AGX Xavier端到端(包含前后处理) 推理 速 度 达到 80FPS 。 同时,支持 多路视频 流部署能力,实现 Jetson AGX Xavier 4路视频流 端到端 20FPS 实时部署。 了解更多
PP-Vehicle:车辆分析工具箱,新增逆行、压线分析、推理提速、多路视频流支持
PP-Vehicle是一款针对车辆分析相关场景的开源工具,产品主要围绕以下几个方面进行设计开发:
实用性
泛化性
低代码
PP-Vehicle车辆分析工具箱新增车辆 逆行分析、压线分析 功能 在 PP-YOLOE+_t 模型的加持下,开源 边缘端实时检测模型 ,精度mAP 53.5, Jetson AGX Xavier 上 端到端(包含前后处理) 推理速度达速度 80FPS 同时,支持 多路视频流 部署能力,实现 Jetson AGX Xavier 4路 视频流端到端 20FPS 实时部署 了解更多 https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.6/deploy/pipeline
三、半监督检测、少样本学习、模型蒸馏算法全面开源
GEEK TALK
除了上面的PP-YOLOE+针对半监督、少样本学习、模型蒸馏方案外,我们还开源了更多的相关模型,大家可以基于不同的模型进行相关方案测试。
半监督目标检测是同时使用有标注数据和无标注数据进行训练的目标检测,既可以极大地节省标注成本,也可以充分利用无标注数据进一步提高检测精度。PaddleDetection团队复现了DenseTeacher半监督检测算法,用户可以下载使用。
了解更多
https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.6/configs/semi_det
跨领域Label Co-tuning算法
领域内Contrastive Tuning算法
了解更多
模型蒸馏
FGD蒸馏算法
FGD全称为Focal and Global Knowledge Distillation for Detectors,是目标检测任务的一种蒸馏方法。FGD蒸馏分为两个部分Focal和Global。Focal蒸馏分离图像的前景和背景,让学生模型分别关注教师模型的前景和背景部分特征的关键像素。Global蒸馏部分重建不同像素之间的关系并将其从教师转移到学生,以补偿Focal蒸馏中丢失的全局信息。实验结果表明,FGD蒸馏算法在基于anchor和anchor free的方法上能有效提升模型精度。在PaddleDetection中,我们实现了FGD算法,并基于RetinaNet算法进行了验证。
CWD蒸馏算法
LD蒸馏算法
了解更多
四、前沿算法量大管饱
GEEK TALK
PaddleDetection持续跟踪深度学习视觉前沿算法,并评估挑选价值比较高的算法进行优化开源。在本次版本发新中新增如下方向算法:
YOLO 家族新增 YOLOv8、YOLOv6-3.0 目标检测算法新增 DINO, YOLOF 新增 ViTDet 系列检测模型, P P-YOLOE+ViT_base、Mask RCNN + ViT_base、Mask RCNN + ViT_large 新增 多目标跟踪 算法 CenterTrack 新增 旋转框检测 算法 FCOSR ; 新增 实例分割 算法 QueryInst ; 新增 3D关键点检测 算法 Metro3d 。 了解更多
https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.6
五、基础能力提升
GEEK TALK
在本次更新中,PaddleDetection支持Python3.10版本、EMA支持过滤不更新参数、AdamW适配飞桨框架2.4.1版本。
新增检测热力图可视化能力,适配FasterRCNN/MaskRCNN系列、PP-YOLOE系列、BlazeFace、SSD、 RetinaNet。
六、Roadmap of PaddleDetection
GEEK TALK
我们已经在PaddleDetection Issue区开辟了Roadmap专区,欢迎您在此留下宝贵的建议、新特性需求或者任何想对PaddleDetection团队说的话,我们会积极融入到后续的版本更新中,同时也欢迎大家参与到PaddleDetection的开源建设中。
Roadmap专区
https://github.com/PaddlePaddle/PaddleDetection/issues/789
2023飞桨黑客马拉松PaddlePaddle Hackathon第四期现已正式启动,本次活动由深度学习技术及应用国家工程研究中心主办,飞桨承办,英特尔作为顶级赞助方,OpenVINO、Arm、Hugging Face、Jina、TVM、瑞芯微、算能、地平线等共同出品,旨在鼓励开发者了解与参与深度学习开源项目的共建。
本次活动上线200+个开发任务,从框架开发、模型套件开发、模型复现、硬件项目开发到产业落地方案应有尽有,根据开发难度分为基础任务和进阶任务。同时提供高达100w的现金奖池以及飞桨黑客松证书,欢迎大家来PaddleDetection专区领取任务:
https://github.com/PaddlePaddle/Paddle/issues/51281#detection
点击阅读原文,体验PaddleDetection最新版本!
END