Hackathon YOLOv8 training

【飞桨黑客松 Hackathon + No.155】Yolov8 模型训练复现
PeiyuLau · May 18, 2023 · 8fbce85 · 8fbce85
2 parents 2dae336 + 048bc3f
commit 8fbce85
Show file tree

Hide file tree

Showing 8 changed files with 586 additions and 142 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
@@ -34,7 +34,7 @@ YOLOv8Head:
     beta: 6.0
   nms:
     name: MultiClassNMS
-    nms_top_k: 1000
+    nms_top_k: 3000
     keep_top_k: 300
     score_threshold: 0.001
     nms_threshold: 0.7
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))