distill

PPYOLOE+ Distillation(PPYOLOE+ 蒸馏)

PaddleDetection提供了对PPYOLOE+ 进行模型蒸馏的方案，结合了logits蒸馏和feature蒸馏。更多蒸馏方案可以查看slim/distill。

模型库

模型	方案	输入尺寸	epochs	Box mAP	配置文件	下载链接
PP-YOLOE+_x	teacher	640	80e	54.7	config	model
PP-YOLOE+_l	student	640	80e	52.9	config	model
PP-YOLOE+_l	distill	640	80e	54.0(+1.1)	config,slim_config	model
PP-YOLOE+_l	teacher	640	80e	52.9	config	model
PP-YOLOE+_m	student	640	80e	49.8	config	model
PP-YOLOE+_m	distill	640	80e	51.0(+1.2)	config,slim_config	model

快速开始

训练

# 单卡
python tools/train.py -c configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml --slim_config configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml
# 多卡
python -m paddle.distributed.launch --log_dir=ppyoloe_plus_distill_x_distill_l/ --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml --slim_config configs/slim/distill/ppyoloe_plus_distill_x_distill_l.yml

• -c: 指定模型配置文件，也是student配置文件。
• --slim_config: 指定压缩策略配置文件，也是teacher配置文件。

评估

python tools/eval.py -c configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml -o weights=output/ppyoloe_plus_crn_l_80e_coco_distill/model_final.pdparams

• -c: 指定模型配置文件，也是student配置文件。
• --slim_config: 指定压缩策略配置文件，也是teacher配置文件。
• -o weights: 指定压缩算法训好的模型路径。

测试

python tools/infer.py -c configs/ppyoloe/distill/ppyoloe_plus_crn_l_80e_coco_distill.yml -o weights=output/ppyoloe_plus_crn_l_80e_coco_distill/model_final.pdparams --infer_img=demo/000000014439_640x640.jpg

• -c: 指定模型配置文件。
• --slim_config: 指定压缩策略配置文件。
• -o weights: 指定压缩算法训好的模型路径。
• --infer_img: 指定测试图像路径。