fix det config (#2068)

docs/tutorials/models/support_model_list.md

@@ -144,9 +144,9 @@
 | PP-YOLOE_plus-M | [PP-YOLOE_plus-M.yaml](../../../paddlex/configs/object_detection/PP-YOLOE_plus-M.yaml)|
 | PP-YOLOE_plus-L | [PP-YOLOE_plus-L.yaml](../../../paddlex/configs/object_detection/PP-YOLOE_plus-L.yaml)|
 | PP-YOLOE_plus-X | [PP-YOLOE_plus-X.yaml](../../../paddlex/configs/object_detection/PP-YOLOE_plus-X.yaml)|
-| PP-YOLOE+_SOD-L | [PP-YOLOE+_SOD-L.yaml](../../../paddlex/configs/smallobject_detection/PP-YOLOE+_SOD-L.yaml)|
-| PP-YOLOE+_SOD-S | [PP-YOLOE+_SOD-S.yaml](../../../paddlex/configs/smallobject_detection/PP-YOLOE+_SOD-S.yaml)|
-| PP-YOLOE+_SOD-largesize-L | [PP-YOLOE+_SOD-largesize-L.yaml](../../../paddlex/configs/smallobject_detection/PP-YOLOE+_SOD-largesize-L.yaml)|
+| PP-YOLOE+_SOD-L | [PP-YOLOE+_SOD-L.yaml](../../../paddlex/configs/small_object_detection/PP-YOLOE+_SOD-L.yaml)|
+| PP-YOLOE+_SOD-S | [PP-YOLOE+_SOD-S.yaml](../../../paddlex/configs/small_object_detection/PP-YOLOE+_SOD-S.yaml)|
+| PP-YOLOE+_SOD-largesize-L | [PP-YOLOE+_SOD-largesize-L.yaml](../../../paddlex/configs/small_object_detection/PP-YOLOE+_SOD-largesize-L.yaml)|
 ### 2. RT-DETR 系列
 | 模型名称 | config |
 | :--- | :---: |

paddlex/configs/anomaly_detection/STFPM.yaml

@@ -1,7 +1,7 @@
 Global:
   model: STFPM
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/mnt/yys/dataset/mv_dataset"
+  dataset_dir: "/paddle/dataset/paddlex/seg/mvtec_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   epochs_iters: 50
   num_classes: 1
   batch_size: 1
-  learning_rate: #0.01
+  learning_rate: 0.4
   pretrain_weight_path: null
-  warmup_steps: #0
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 100
@@ -38,4 +38,4 @@ Predict:
 
   
 Export:
-  weight_path: https://bj.bcebos.com/paddleseg/dygraph/mvtec_ad/stfpm/model.pdparams
+  weight_path: https://paddleseg.bj.bcebos.com/dygraph/mvtec_ad/stfpm/stfpm.pdparams

paddlex/configs/face_detection/PicoDet_LCNet_x2_5_face.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PicoDet_LCNet_x2_5_face
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/widerface_coco_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 4
-  learning_rate: #0.08
+  learning_rate: 0.08
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/human_detection/PP-YOLOE-L_human.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE-L_human
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/widerperson_coco_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/human_detection/PP-YOLOE-S_human.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE-S_human
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/widerperson_coco_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/mainbody_detection/PP-ShiTuV2_det.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-ShiTuV2_det
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/mainbody_det_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.4
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/Cascade-FasterRCNN-ResNet50-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/CenterNet-DLA-34.yaml

@@ -18,7 +18,7 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: 0.08
+  learning_rate: 0.0005
   pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null

paddlex/configs/object_detection/CenterNet-ResNet50.yaml

@@ -18,7 +18,7 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: 0.08
+  learning_rate: 0.0005
   pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null

paddlex/configs/object_detection/DETR-R50.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.0001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FCOS-ResNet50.yaml

@@ -18,7 +18,7 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 4
-  learning_rate: 0.002
+  learning_rate: 0.01
   pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null

paddlex/configs/object_detection/FasterRCNN-ResNeXt101-vd-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet101-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet101.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet34-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet50-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet50-vd-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet50-vd-SSLDv2-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-ResNet50.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 5
   eval_interval: 1

paddlex/configs/object_detection/FasterRCNN-Swin-Tiny-FPN.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.0001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/object_detection/PicoDet-M.yaml

@@ -18,7 +18,7 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: 0.08
+  learning_rate: 0.24
   pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null

paddlex/configs/object_detection/PicoDet-XS.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.28
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/smallobject_detection/PP-YOLOE_plus_SOD-L.yaml → paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-L.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE_plus_SOD-L
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/small_det_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: 0.001
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/smallobject_detection/PP-YOLOE_plus_SOD-S.yaml → paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-S.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE_plus_SOD-S
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/small_det_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: 0.001
+  learning_rate: 0.01
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/smallobject_detection/PP-YOLOE_plus_SOD-largesize-L.yaml → paddlex/configs/small_object_detection/PP-YOLOE_plus_SOD-largesize-L.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE_plus_SOD-largesize-L
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/small_det_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,7 +18,7 @@ Train:
   num_classes: 4
   epochs_iters: 10
   batch_size: 2
-  learning_rate: 0.0001
+  learning_rate: 0.00125
   pretrain_weight_path: null
   warmup_steps: 100
   resume_path: null

paddlex/configs/structure_analysis/PicoDet-L_layout_3cls.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 11
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.4
+  learning_rate: 0.06
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/structure_analysis/RT-DETR-H_layout_17cls.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 11
   epochs_iters: 50
   batch_size: 1
-  learning_rate: 0.00005
+  learning_rate: 0.0001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/structure_analysis/RT-DETR-H_layout_3cls.yaml

@@ -18,9 +18,9 @@ Train:
   num_classes: 11
   epochs_iters: 50
   batch_size: 1
-  learning_rate: 0.00005
+  learning_rate: 0.0001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/vehicle_detection/PP-YOLOE-L_vehicle.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE-L_vehicle
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/vehicle_coco_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/configs/vehicle_detection/PP-YOLOE-S_vehicle.yaml

@@ -1,7 +1,7 @@
 Global:
   model: PP-YOLOE-S_vehicle
   mode: check_dataset # check_dataset/train/evaluate/predict
-  dataset_dir: "/paddle/dataset/paddlex/det/det_coco_examples"
+  dataset_dir: "/paddle/dataset/paddlex/det/vehicle_coco_examples"
   device: gpu:0,1,2,3
   output: "output"
 
@@ -18,9 +18,9 @@ Train:
   num_classes: 4
   epochs_iters: 50
   batch_size: 2
-  learning_rate: #0.08
+  learning_rate: 0.001
   pretrain_weight_path: null
-  warmup_steps: #100
+  warmup_steps: 100
   resume_path: null
   log_interval: 10
   eval_interval: 1

paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-L.yaml

@@ -129,7 +129,7 @@ PPYOLOEHead:
     beta: 6.0
   nms:
     name: MultiClassNMS
-    nms_top_k: 1000
+    nms_top_k: 10000
     keep_top_k: 500
     score_threshold: 0.01
     nms_threshold: 0.6

paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-S.yaml

@@ -129,7 +129,7 @@ PPYOLOEHead:
     beta: 6.0
   nms:
     name: MultiClassNMS
-    nms_top_k: 1000
+    nms_top_k: 10000
     keep_top_k: 500
     score_threshold: 0.01
     nms_threshold: 0.6

paddlex/repo_apis/PaddleDetection_api/configs/PP-YOLOE_plus_SOD-largesize-L.yaml

@@ -131,7 +131,7 @@ PPYOLOEHead:
     beta: 6.0
   nms:
     name: MultiClassNMS
-    nms_top_k: 1000
+    nms_top_k: 10000
     keep_top_k: 500
     score_threshold: 0.01
     nms_threshold: 0.6