精度54.7mAP，相較YOLOv7提升1.9%

L版本端到端推理速度42.2FPS

訓練速度提升3.75倍

COCO數(shù)據(jù)集僅需20epoch即可達到50.0mAP

下游任務泛化性最高提升8%

10+即開即用多端部署Demo

這究竟是什么模型，竟可達到以上效果？答案是：PP-YOLOE+PP-YOLOE+是基于飛槳云邊一體高精度模型PP-YOLOE迭代優(yōu)化升級的版本，具備以下特點：

• 超強性能

表格1：PP-YOLOE+與PP-YOLOE性能對比*備注：以上速度為端到端推理速度，包含數(shù)據(jù)解碼+數(shù)據(jù)預處理+模型預測+后處理計算；均在v100上測試所得，V100 + CUDA11.2 + cudnn8.2.0 + TRT8.0.1.6

• 訓練收斂加速：使用Objects365預訓練模型，減少訓練輪數(shù)，訓練收斂速度提升3.75倍。
• 下游任務泛化性顯著提升：在農(nóng)業(yè)、夜間安防、工業(yè)等不同場景數(shù)據(jù)集上驗證，精度最高提升8.1%。

• 高性能部署能力：本次升級PP-YOLOE+支持多種部署方式，包括Python/C++、Serving、ONNX Runtime、ONNX-TRT、INT8量化等部署能力。

超強性能與超高泛化性使得PP-YOLOE+助力開發(fā)者在最短時間、最少量數(shù)據(jù)上能得到最優(yōu)效果。模型下載與完整教程請見PP-YOLOE+：https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/configs/ppyoloe 

回顧：PP-YOLOE經(jīng)典網(wǎng)絡結構剖析

圖1：PP-YOLOE網(wǎng)絡結構PP-YOLOE是一款高精度推理速度快的檢測模型，包含骨干網(wǎng)絡CSPRepResNet、特征融合CSPPAN、輕量級ET-Head和改進的動態(tài)匹配算法TAL（Task Alignment Learning）等模塊， 并且根據(jù)不同的應用場景設計了一系列模型，即s/m/l/x。PP-YOLOE的全系列模型從精度速度性價比來看達到工業(yè)界最優(yōu)。具體來說，PP-YOLOE-l在COCO test-dev上AP可達51.4，在V100上速度可達78.1 FPS，使用TensorRT FP16進行推理，在V100上速度可達149FPS。PP-YOLOE-l相較于PP-YOLOv2精度提升1.9AP、速度提升13.35%，相對于YOLOX-l精度提升1.3AP、速度提升24.96%。此外，PP-YOLOE還避免使用諸如可變形卷積或者Matrix NMS之類的特殊算子，使PP-YOLOE全系列模型能輕松地部署在NVIDIA V100和T4這樣的云端GPU架構、Jetson系列的移動端GPU和高性能的FPGA開發(fā)板上。具體的結構包括以下三大部分：

可擴展的backbone和neck

我們設計了CSPRepResNet作為backbone, neck部分也采用了新設計的CSPPAN結構，backbone和neck均以我們提出的CSPRepResStage為基礎模塊。新的backbone和neck在增強模型表征能力的同時提升了模型的推理速度，并且可以通過width multiplier和depth multiplier靈活地配置模型的大小。

TAL（Task Alignment Learning）

為了進一步提升模型的精度，我們選用了TOOD [1]中的動態(tài)匹配算法策略TAL。TAL同時考慮分類和回歸，使得匹配結果同時獲得了最優(yōu)的分類和定位精度。

Efficient Task-aligned head

• 升級一：強大的Objects365預訓練模型、升級版backbone等改動大幅提升PP-YOLOE系列模型的精度；

• 升級二：優(yōu)化預處理，提升模型端到端推理速度，更貼近用戶使用的真實場景；

• 升級三：完善多種環(huán)境下的推理部署能力。

精度

其次，我們在RepResBlock中的1x1卷積上增加了一個可學習的權重alpha，進一步提升了backbone的表征能力，獲得了不錯的效果提升。最后，我們調(diào)整了NMS的參數(shù)，在COCO上可以獲得更好的評估精度。

訓練速度

基于Objects365的預訓練模型，將學習率調(diào)整為原始學習率的十分之一，訓練的epoch從300降到了80，在大大縮短了訓練時間的同時，獲得了精度上的提升。

端到端推理速度

圖2：PP-YOLOE+模型效果

下游泛化性增強

• 鏈接

https://github.com/thsant/wgisd低光數(shù)據(jù)集使用ExDark，該數(shù)據(jù)集是一個專門在低光照環(huán)境下拍攝出針對低光目標檢測的數(shù)據(jù)集，包括從極低光環(huán)境到暮光環(huán)境等10種不同光照條件下的圖片。

• 鏈接

https://github.com/cs-chan/Exclusively-Dark-Image-Dataset工業(yè)數(shù)據(jù)集使用PKU-Market-PCB，該數(shù)據(jù)集用于印刷電路板（PCB）的瑕疵檢測，提供了6種常見的PCB缺陷。

• 鏈接

https://robotics.pkusz.edu.cn/resources/dataset/

精度

表格2：PP-YOLOE+在不同場景數(shù)據(jù)集的效果

可視化效果

全面升級的部署支持

• 文檔鏈接

https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.5/deploy總的來說，PP-YOLOE+希望真正的站在用戶的角度去解決訓練慢、泛化性差、部署難的問題。通過使用大規(guī)模數(shù)據(jù)集Objects365對模型進行預訓練，使得模型在下游任務上表現(xiàn)更魯棒、泛化能力更強。通過優(yōu)化推理過程，完善推理鏈條，使得PP-YOLOE+不但在端到端預測速度上得到極大的提升，還拓展了Python、C++、Serving、ONNX Runtime以及TensorRT等多種推理環(huán)境下的部署能力，提升模型可擴展性，從而提升用戶的使用體驗。

飛槳端到端開發(fā)套件PaddleDetection


PaddleDetection為基于飛槳的端到端目標檢測套件，內(nèi)置30+模型算法及300+預訓練模型，覆蓋目標檢測、實例分割、跟蹤、關鍵點檢測等方向，其中包括服務器端和移動端高精度、輕量化產(chǎn)業(yè)級SOTA模型、冠軍方案和學術前沿算法，并提供即插即用的垂類場景預訓練模型，覆蓋人、車等20+場景，提供配置化的網(wǎng)絡模塊組件、十余種數(shù)據(jù)增強策略和損失函數(shù)等高階優(yōu)化支持和多種部署方案，在打通數(shù)據(jù)處理、模型開發(fā)、訓練、壓縮、部署全流程的基礎上，提供豐富的案例及教程，加速算法產(chǎn)業(yè)落地應用。圖4：PaddleDetection全景圖

• 項目鏈接

https://github.com/PaddlePaddle/PaddleDetection

• 參考文獻

[1] Feng C , Zhong Y , Gao Y , et al. TOOD: Task-aligned One-stage Object Detection[C]// 2021