yolov8 modify

PeiyuLau · May 14, 2023 · 0002ef5 · 0002ef5
1 parent 4546815
commit 0002ef5
Show file tree

Hide file tree

Showing 8 changed files with 602 additions and 126 deletions.
diff --git a/README.md b/README.md
diff --git a/README.md b/README.md
diff --git a/configs/yolov8/_base_/optimizer_500e.yml b/configs/yolov8/_base_/optimizer_500e.yml
@@ -17,3 +17,4 @@ OptimizerBuilder:
   regularizer:
     factor: 0.0005
     type: L2
+  clip_grad_by_value: 10.
diff --git a/configs/yolov8/_base_/optimizer_500e_high.yml b/configs/yolov8/_base_/optimizer_500e_high.yml
@@ -17,3 +17,4 @@ OptimizerBuilder:
   regularizer:
     factor: 0.0005
     type: L2
+  clip_grad_by_value: 10.
diff --git a/configs/yolov8/_base_/yolov8_cspdarknet.yml b/configs/yolov8/_base_/yolov8_cspdarknet.yml
@@ -1,5 +1,5 @@
 architecture: YOLOv8
-norm_type: sync_bn
+#norm_type: sync_bn
 use_ema: True
 ema_decay: 0.9999
 ema_decay_type: "exponential"

diff --git a/configs/yolov8/_base_/yolov8_reader.yml b/configs/yolov8/_base_/yolov8_reader.yml
@@ -29,7 +29,7 @@ EvalReader:
     - Pad: {size: *input_size, fill_value: [114., 114., 114.]}
     - NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
     - Permute: {}
-  batch_size: 1
+  batch_size: 8
 
 
 TestReader:

diff --git a/configs/yolov8/_base_/yolov8_reader_high_aug.yml b/configs/yolov8/_base_/yolov8_reader_high_aug.yml
@@ -7,7 +7,7 @@ worker_num: 4
 TrainReader:
   sample_transforms:
     - Decode: {}
-    - MosaicPerspective: {mosaic_prob: 1.0, target_size: *input_size, scale: 0.9, mixup_prob: 0.15}
+    - MosaicPerspective: {mosaic_prob: 1.0, target_size: *input_size, scale: 0.9, mixup_prob: 0.1, copy_paste_prob: 0.1}
     - RandomHSV: {hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - RandomFlip: {}
   batch_transforms:
@@ -29,7 +29,7 @@ EvalReader:
     - Pad: {size: *input_size, fill_value: [114., 114., 114.]}
     - NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
     - Permute: {}
-  batch_size: 1
+  batch_size: 8
 
 
 TestReader:

diff --git a/ppdet/engine/trainer.py b/ppdet/engine/trainer.py
@@ -69,8 +69,7 @@ def __init__(self, cfg, mode='train'):
         self.custom_white_list = self.cfg.get('custom_white_list', None)
         self.custom_black_list = self.cfg.get('custom_black_list', None)
 
-        if self.cfg.architecture in ['RTMDet', 'YOLOv6', 'YOLOv8'
-                                     ] and self.mode == 'train':
+        if self.cfg.architecture in ['RTMDet', 'YOLOv6'] and self.mode == 'train':
             raise NotImplementedError('{} training not supported yet.'.format(
                 self.cfg.architecture))