From ac701833d3421d5f0ffc88b1dc6df6fc5053e61b Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Thu, 15 Apr 2021 13:58:28 +0800
Subject: [PATCH] fix yolo config (#2635)

---
 configs/ppyolo/ppyolo_r50vd_dcn_voc.yml     | 14 +++++++-------
 configs/slim/prune/yolov3_prune_fpgm.yml    |  1 -
 configs/slim/prune/yolov3_prune_l1_norm.yml |  1 -
 3 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/configs/ppyolo/ppyolo_r50vd_dcn_voc.yml b/configs/ppyolo/ppyolo_r50vd_dcn_voc.yml
index 4b2bcc492..b383deab0 100644
--- a/configs/ppyolo/ppyolo_r50vd_dcn_voc.yml
+++ b/configs/ppyolo/ppyolo_r50vd_dcn_voc.yml
@@ -11,13 +11,13 @@ weights: output/ppyolo_r50vd_dcn_voc/model_final
 
 TrainReader:
   batch_transforms:
-    - BatchRandomResizeOp: {target_size: [320, 352, 384, 416, 448, 480, 512, 544, 576, 608], random_size: True, random_interp: True, keep_ratio: False}
-    - NormalizeBoxOp: {}
-    - PadBoxOp: {num_max_boxes: 50}
-    - BboxXYXY2XYWHOp: {}
-    - NormalizeImageOp: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
-    - PermuteOp: {}
-    - Gt2YoloTargetOp: {anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]], anchors: [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]], downsample_ratios: [32, 16, 8], num_classes: 20}
+    - BatchRandomResize: {target_size: [320, 352, 384, 416, 448, 480, 512, 544, 576, 608], random_size: True, random_interp: True, keep_ratio: False}
+    - NormalizeBox: {}
+    - PadBox: {num_max_boxes: 50}
+    - BboxXYXY2XYWH: {}
+    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
+    - Permute: {}
+    - Gt2YoloTarget: {anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]], anchors: [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]], downsample_ratios: [32, 16, 8], num_classes: 20}
   mixup_epoch: 350
   batch_size: 12
 
diff --git a/configs/slim/prune/yolov3_prune_fpgm.yml b/configs/slim/prune/yolov3_prune_fpgm.yml
index 8d99997ef..f37453868 100644
--- a/configs/slim/prune/yolov3_prune_fpgm.yml
+++ b/configs/slim/prune/yolov3_prune_fpgm.yml
@@ -1,6 +1,5 @@
 # Weights of yolov3_mobilenet_v1_voc
 pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_voc.pdparams
-weight_type: resume
 slim: Pruner
 
 Pruner:
diff --git a/configs/slim/prune/yolov3_prune_l1_norm.yml b/configs/slim/prune/yolov3_prune_l1_norm.yml
index 757117e38..5b4f4667f 100644
--- a/configs/slim/prune/yolov3_prune_l1_norm.yml
+++ b/configs/slim/prune/yolov3_prune_l1_norm.yml
@@ -1,6 +1,5 @@
 # Weights of yolov3_mobilenet_v1_voc
 pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_voc.pdparams
-weight_type: resume
 slim: Pruner
 
 Pruner:
-- 
GitLab