1 Star 0 Fork 13

付肖威 / PaddleSlim

forked from PaddlePaddle / PaddleSlim 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
paddledetection_slim_distillation_tutorial.md 2.11 KB
一键复制 编辑 原始数据 按行查看 历史
恬淡虚无 提交于 2022-05-05 10:50 . update docs (#1071)

目标检测模型蒸馏教程

教程内容请参考:https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.4/static/slim/distillation/README.md

示例结果

MobileNetV1-YOLO-V3-VOC

FLOPS 输入尺寸 每张GPU图片个数 推理时间(fps) Box AP 下载
baseline 608 16 104.291 76.2 下载链接
蒸馏后 608 16 106.914 79.0 下载链接
baseline 416 16 - 76.7 下载链接
蒸馏后 416 16 - 78.2 下载链接
baseline 320 16 - 75.3 下载链接
蒸馏后 320 16 - 75.5 下载链接

蒸馏后的结果用ResNet34-YOLO-V3做teacher,4GPU总batch_size64训练90000 iter得到

MobileNetV1-YOLO-V3-COCO

FLOPS 输入尺寸 每张GPU图片个数 推理时间(fps) Box AP 下载
baseline 608 16 78.302 29.3 下载链接
蒸馏后 608 16 78.523 31.4 下载链接
baseline 416 16 - 29.3 下载链接
蒸馏后 416 16 - 30.0 下载链接
baseline 320 16 - 27.0 下载链接
蒸馏后 320 16 - 27.1 下载链接

蒸馏后的结果用ResNet34-YOLO-V3做teacher,4GPU总batch_size64训练600000 iter得到

Python
1
https://gitee.com/fu_xiao_wei/PaddleSlim.git
git@gitee.com:fu_xiao_wei/PaddleSlim.git
fu_xiao_wei
PaddleSlim
PaddleSlim
develop

搜索帮助