diff --git a/docs/images/PULC/docs/safety_helmet_data_demo.png b/docs/images/PULC/docs/safety_helmet_data_demo.png new file mode 100644 index 0000000000000000000000000000000000000000..8e2f1c4afe460bf2e0867c3bd35dc22fd3211293 Binary files /dev/null and b/docs/images/PULC/docs/safety_helmet_data_demo.png differ diff --git a/docs/zh_CN/PULC/PULC_safety_helmet_cls.md b/docs/zh_CN/PULC/PULC_safety_helmet_cls.md index 91bc860876e7387f7b48c02aaab6317c5cf9fe4e..bb47c289a978be644ac6b5442e3ff8da8247a344 100644 --- a/docs/zh_CN/PULC/PULC_safety_helmet_cls.md +++ b/docs/zh_CN/PULC/PULC_safety_helmet_cls.md @@ -44,6 +44,7 @@ | 模型 | Tpr(%) | 延时(ms) | 存储(M) | 策略 | |-------|-----------|----------|---------------|---------------| +| Res2Net200_vd_26w_4s | 98.92 | - | - | 使用ImageNet预训练模型 | | SwinTranformer_tiny | 93.57 | 175.52 | 107 | 使用ImageNet预训练模型 | | MobileNetV3_large_x1_0 | 97.47 | 4.70 | 17 | 使用ImageNet预训练模型 | | PPLCNet_x1_0 | 93.29 | 2.36 | 6.5 | 使用ImageNet预训练模型 | @@ -52,7 +53,9 @@ | PPLCNet_x1_0 | 98.71 | 2.36 | 6.5 | 使用SSLD预训练模型+EDA策略+SKL-UGI知识蒸馏策略| -从表中可以看出,backbone 为 SwinTranformer_tiny 时精度较高,但是推理速度较慢。将 backboone 替换为轻量级模型 MobileNetV3_large_x1_0 后,速度可以大幅提升,但是精度下降明显。将 backbone 替换为 PPLCNet_x1_0 时,精度较 MobileNetV3_large_x1_0 低两个多百分点,但是速度提升 2 倍左右。在此基础上,使用 SSLD 预训练模型后,在不改变推理速度的前提下,精度可以提升约 2.6 个百分点,进一步地,当融合EDA策略后,精度可以再提升 1.3 个百分点,最后,在使用 SKL-UGI 知识蒸馏后,精度可以继续提升 2.2 个百分点。此时,PPLCNet_x1_0 达到了 SwinTranformer_tiny 模型的精度,但是速度快 70+ 倍。关于 PULC 的训练方法和推理部署方法将在下面详细介绍。 +从表中可以看出,backbone 为 Res2Net200_vd_26w_4s 时精度较高,但是推理速度较慢。将 backboone 替换为轻量级模型 MobileNetV3_large_x1_0 后,速度可以大幅提升,但是精度下降明显。将 backbone 替换为 PPLCNet_x1_0 时,精度较 MobileNetV3_large_x1_0 低四个多百分点,但是速度提升 2 倍左右。在此基础上,使用 SSLD 预训练模型后,在不改变推理速度的前提下,精度可以提升约 4.8 个百分点,进一步地,当融合EDA策略后,精度可以再提升 0.7 个百分点。此时,PPLCNet_x1_0 已经接近了 Res2Net200_vd_26w_4s 模型的精度,但是速度快 70+ 倍。 + + **备注:** @@ -120,16 +123,11 @@ cd ../ 执行上述命令后,`dataset/` 下存在 `safety_helmet` 目录,该目录中具有以下数据: - - ``` -├── train -│   ├── 000000000009.jpg -│   ├── 000000000025.jpg -... -├── val -│   ├── objects365_01780637.jpg -│   ├── objects365_01780640.jpg +├── images +│   ├── VOC2028_part2_001209_1.jpg +│   ├── HHD_hard_hat_workers23_1.jpg +│   ├── CelebA_077809.jpg ... ├── ImageNet_val │   ├── ILSVRC2012_val_00000001.JPEG