Procházet zdrojové kódy

set use_ema for YOLOv3 and YOLOX models (#1932)

* fix bug

* Update YOLOX-M.yaml

* Update YOLOX-N.yaml

* Update YOLOX-S.yaml

* Update YOLOX-T.yaml

* Update YOLOX-X.yaml

* Update YOLOv3-DarkNet53.yaml

* Update YOLOv3-MobileNetV3.yaml

* Update YOLOv3-ResNet50_vd_DCN.yaml
Liu Jiaxuan před 1 rokem
rodič
revize
cc1c50c15c

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-L.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -147,4 +148,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-M.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -147,4 +148,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-N.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -142,4 +143,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-S.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -147,4 +148,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-T.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -141,4 +142,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOX-X.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 100
 save_dir: output
 snapshot_epoch: 10
@@ -147,4 +148,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOv3-DarkNet53.yaml

@@ -2,6 +2,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 20
 save_dir: output
 snapshot_epoch: 5
@@ -149,4 +150,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 2
paddlex/repo_apis/PaddleDetection_api/configs/YOLOv3-MobileNetV3.yaml

@@ -4,7 +4,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
-log_iter: 20
+use_ema: True
 save_dir: output
 snapshot_epoch: 1
 print_flops: false
@@ -150,4 +150,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False

+ 2 - 1
paddlex/repo_apis/PaddleDetection_api/configs/YOLOv3-ResNet50_vd_DCN.yaml

@@ -4,6 +4,7 @@ use_gpu: true
 use_xpu: false
 use_mlu: false
 use_npu: false
+use_ema: True
 log_iter: 20
 save_dir: output
 snapshot_epoch: 1
@@ -151,4 +152,4 @@ export:
   post_process: True  # Whether post-processing is included in the network when export model.
   nms: True           # Whether NMS is included in the network when export model.
   benchmark: False    # It is used to testing model performance, if set `True`, post-process and NMS will not be exported.
-  fuse_conv_bn: False
+  fuse_conv_bn: False