更换文档检测模型

2024-08-27 14:42:45 +08:00
parent aea6f19951
commit 1514e09c40
2072 changed files with 254336 additions and 4967 deletions
--- a/paddle_detection/configs/slim/prune/faster_rcnn_r50_fpn_prune_fpgm.yml
+++ b/paddle_detection/configs/slim/prune/faster_rcnn_r50_fpn_prune_fpgm.yml
@@ -0,0 +1,16 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_fpn_1x_coco.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: fpgm
+  pruned_params: ['conv2d_27.w_0', 'conv2d_28.w_0', 'conv2d_29.w_0',
+                  'conv2d_30.w_0', 'conv2d_31.w_0', 'conv2d_32.w_0',
+                  'conv2d_33.w_0', 'conv2d_34.w_0', 'conv2d_35.w_0',
+                  'conv2d_36.w_0', 'conv2d_37.w_0', 'conv2d_38.w_0',
+                  'conv2d_39.w_0', 'conv2d_40.w_0', 'conv2d_41.w_0',
+                  'conv2d_42.w_0', 'conv2d_43.w_0', 'conv2d_44.w_0',
+                  'conv2d_45.w_0', 'conv2d_46.w_0', 'conv2d_47.w_0',
+                  'conv2d_48.w_0', 'conv2d_49.w_0', 'conv2d_50.w_0',
+                  'conv2d_51.w_0', 'conv2d_52.w_0']
+  pruned_ratios: [0.1,0.2,0.2,0.2,0.2,0.1,0.2,0.3,0.3,0.3,0.2,0.1,0.3,0.4,0.4,0.4,0.4,0.3,0.4,0.4,0.4,0.4,0.4,0.4,0.4,0.4]
+  print_params: False
--- a/paddle_detection/configs/slim/prune/picodet_m_unstructured_prune_75.yml
+++ b/paddle_detection/configs/slim/prune/picodet_m_unstructured_prune_75.yml
@@ -0,0 +1,11 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/picodet_m_320_coco.pdparams
+slim: UnstructuredPruner
+
+UnstructuredPruner:
+    stable_epochs: 0
+    pruning_epochs: 150
+    tunning_epochs: 150
+    pruning_steps: 300
+    ratio: 0.75
+    initial_ratio: 0.15
+    prune_params_type: conv1x1_only
--- a/paddle_detection/configs/slim/prune/picodet_m_unstructured_prune_85.yml
+++ b/paddle_detection/configs/slim/prune/picodet_m_unstructured_prune_85.yml
@@ -0,0 +1,11 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/picodet_m_320_coco.pdparams
+slim: UnstructuredPruner
+
+UnstructuredPruner:
+    stable_epochs: 0
+    pruning_epochs: 150
+    tunning_epochs: 150
+    pruning_steps: 300
+    ratio: 0.85
+    initial_ratio: 0.20
+    prune_params_type: conv1x1_only
--- a/paddle_detection/configs/slim/prune/ppyolo_mbv3_large_prune_fpgm.yml
+++ b/paddle_detection/configs/slim/prune/ppyolo_mbv3_large_prune_fpgm.yml
@@ -0,0 +1,9 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: fpgm
+  pruned_params: ['conv2d_62.w_0', 'conv2d_63.w_0', 'conv2d_64.w_0',
+                  'conv2d_65.w_0', 'conv2d_66.w_0', 'conv2d_67.w_0']
+  pruned_ratios: [0.75, 0.75, 0.75, 0.75, 0.75, 0.75]
+  print_params: True
--- a/paddle_detection/configs/slim/prune/ppyolo_r50vd_prune_fpgm.yml
+++ b/paddle_detection/configs/slim/prune/ppyolo_r50vd_prune_fpgm.yml
@@ -0,0 +1,13 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: fpgm
+  pruned_params: ['conv2d_56.w_0', 'conv2d_57.w_0', 'conv2d_58.w_0',
+                  'conv2d_59.w_0', 'conv2d_60.w_0', 'conv2d_61.w_0',
+                  'conv2d_63.w_0', 'conv2d_64.w_0', 'conv2d_65.w_0',
+                  'conv2d_66.w_0', 'conv2d_67.w_0', 'conv2d_68.w_0',
+                  'conv2d_70.w_0', 'conv2d_71.w_0', 'conv2d_72.w_0',
+                  'conv2d_73.w_0', 'conv2d_74.w_0', 'conv2d_75.w_0']
+  pruned_ratios: [0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.875,0.875,0.875,0.875,0.875,0.875]
+  print_params: False
--- a/paddle_detection/configs/slim/prune/yolov3_darknet_prune_fpgm.yml
+++ b/paddle_detection/configs/slim/prune/yolov3_darknet_prune_fpgm.yml
@@ -0,0 +1,13 @@
+pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_darknet53_270e_coco.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: fpgm
+  pruned_params: ['conv2d_52.w_0', 'conv2d_53.w_0', 'conv2d_54.w_0',
+                  'conv2d_55.w_0', 'conv2d_56.w_0', 'conv2d_57.w_0',
+                  'conv2d_59.w_0', 'conv2d_60.w_0', 'conv2d_61.w_0',
+                  'conv2d_62.w_0', 'conv2d_63.w_0', 'conv2d_64.w_0',
+                  'conv2d_66.w_0', 'conv2d_67.w_0', 'conv2d_68.w_0',
+                  'conv2d_69.w_0', 'conv2d_70.w_0', 'conv2d_71.w_0']
+  pruned_ratios: [0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.75,0.875,0.875,0.875,0.875,0.875,0.875]
+  print_params: True
--- a/paddle_detection/configs/slim/prune/yolov3_prune_fpgm.yml
+++ b/paddle_detection/configs/slim/prune/yolov3_prune_fpgm.yml
@@ -0,0 +1,14 @@
+# Weights of yolov3_mobilenet_v1_voc
+pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_voc.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: fpgm
+  pruned_params: ['conv2d_27.w_0', 'conv2d_28.w_0', 'conv2d_29.w_0',
+                  'conv2d_30.w_0', 'conv2d_31.w_0', 'conv2d_32.w_0',
+                  'conv2d_34.w_0', 'conv2d_35.w_0', 'conv2d_36.w_0',
+                  'conv2d_37.w_0', 'conv2d_38.w_0', 'conv2d_39.w_0',
+                  'conv2d_41.w_0', 'conv2d_42.w_0', 'conv2d_43.w_0',
+                  'conv2d_44.w_0', 'conv2d_45.w_0', 'conv2d_46.w_0']
+  pruned_ratios: [0.1,0.2,0.2,0.2,0.2,0.1,0.2,0.3,0.3,0.3,0.2,0.1,0.3,0.4,0.4,0.4,0.4,0.3]
+  print_params: False
--- a/paddle_detection/configs/slim/prune/yolov3_prune_l1_norm.yml
+++ b/paddle_detection/configs/slim/prune/yolov3_prune_l1_norm.yml
@@ -0,0 +1,14 @@
+# Weights of yolov3_mobilenet_v1_voc
+pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_voc.pdparams
+slim: Pruner
+
+Pruner:
+  criterion: l1_norm
+  pruned_params: ['conv2d_27.w_0', 'conv2d_28.w_0', 'conv2d_29.w_0',
+                  'conv2d_30.w_0', 'conv2d_31.w_0', 'conv2d_32.w_0',
+                  'conv2d_34.w_0', 'conv2d_35.w_0', 'conv2d_36.w_0',
+                  'conv2d_37.w_0', 'conv2d_38.w_0', 'conv2d_39.w_0',
+                  'conv2d_41.w_0', 'conv2d_42.w_0', 'conv2d_43.w_0',
+                  'conv2d_44.w_0', 'conv2d_45.w_0', 'conv2d_46.w_0']
+  pruned_ratios: [0.1,0.2,0.2,0.2,0.2,0.1,0.2,0.3,0.3,0.3,0.2,0.1,0.3,0.4,0.4,0.4,0.4,0.3]
+  print_params: False