fit new COCODataset

mindspore-lab · Sep 8, 2023 · 58f83f6 · 58f83f6
1 parent 0313ab2
commit 58f83f6
Show file tree

Hide file tree

Showing 14 changed files with 164 additions and 106 deletions.
diff --git a/GETTING_STARTED.md b/GETTING_STARTED.md
@@ -64,7 +64,7 @@ to understand their behavior. Some common arguments are:
   ```
 * To evaluate a model's performance 8 NPUs/GPUs:
   ```
-  mpirun --allow-run-as-root -n 8 python test.py --config ./configs/yolov7/yolov7.yaml --weight /path_to_ckpt/WEIGHT.ckpt --is_parallel True  ```
+  mpirun --allow-run-as-root -n 8 python test.py --config ./configs/yolov7/yolov7.yaml --weight /path_to_ckpt/WEIGHT.ckpt --is_parallel True
   ```
 *Notes: (1) The default hyper-parameter is used for 8-card training, and some parameters need to be adjusted in the case of a single card. (2) The default device is Ascend, and you can modify it by specifying 'device_target' as Ascend/GPU/CPU, as these are currently supported.*
 * For more options, see `train/test.py -h`.

diff --git a/configs/yolov3/hyp.scratch.yaml b/configs/yolov3/hyp.scratch.yaml
@@ -29,19 +29,23 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.0, translate: 0.1, scale: 0.9 }
-    - { func_name: mixup, prob: 0.1, alpha: 8.0, beta: 8.0, needed_mosaic: True }
+    - { func_name: mosaic, prob: 1.0 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }
+    - { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 0.1, pre_transform: [
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }, ]
+    }
     - { func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 }
-    - { func_name: label_norm, xyxy2xywh_: True }
     - { func_name: albumentations }
     - { func_name: fliplr, prob: 0.5 }
+    - { func_name: label_norm, xyxy2xywh_: True }
     - { func_name: label_pad, padding_size: 160, padding_value: -1 }
     - { func_name: image_norm, scale: 255. }
     - { func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
 
   test_transforms:
-    - { func_name: letterbox, scaleup: False }
-    - { func_name: label_norm, xyxy2xywh_: True }
-    - { func_name: label_pad, padding_size: 160, padding_value: -1 }
+    - { func_name: letterbox, scaleup: False, only_image: True }
     - { func_name: image_norm, scale: 255. }
     - { func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov4/hyp.scratch.yaml b/configs/yolov4/hyp.scratch.yaml
@@ -26,19 +26,23 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.0, translate: 0.1, scale: 0.9 }
-    - { func_name: mixup, prob: 0.1, alpha: 8.0, beta: 8.0, needed_mosaic: True }
+    - { func_name: mosaic, prob: 1.0 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }
+    - { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 0.1, pre_transform: [
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }, ]
+    }
     - { func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 }
-    - { func_name: label_norm, xyxy2xywh_: True }
     - { func_name: albumentations }
     - { func_name: fliplr, prob: 0.5 }
+    - { func_name: label_norm, xyxy2xywh_: True }
     - { func_name: label_pad, padding_size: 160, padding_value: -1 }
     - { func_name: image_norm, scale: 255. }
     - { func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
 
   test_transforms:
-    - { func_name: letterbox, scaleup: False }
-    - { func_name: label_norm, xyxy2xywh_: True }
-    - { func_name: label_pad, padding_size: 160, padding_value: -1 }
+    - { func_name: letterbox, scaleup: False, only_image: True }
     - { func_name: image_norm, scale: 255. }
     - { func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov5/hyp.scratch-high.yaml b/configs/yolov5/hyp.scratch-high.yaml
@@ -33,19 +33,25 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - {func_name: mosaic, prob: 1.0, mosaic9_prob: 0.0, translate: 0.1, scale: 0.9, copy_paste_prob: 0.1}
-    - {func_name: mixup, prob: 0.1, alpha: 32.0, beta: 32.0, needed_mosaic: True}
-    - {func_name: label_norm, xyxy2xywh_: True}
+    - { func_name: mosaic, prob: 1.0 }
+    - { func_name: copy_paste, prob: 0.1 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }
+    - { func_name: mixup, alpha: 32.0, beta: 32.0, prob: 0.1, pre_transform: [
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: copy_paste, prob: 0.1 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }, ]
+    }
     - {func_name: albumentations}
     - {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - {func_name: fliplr, prob: 0.5}
+    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: label_pad, padding_size: 160, padding_value: -1}
     - {func_name: image_norm, scale: 255.}
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
 
   test_transforms:
-    - {func_name: letterbox, scaleup: False}
-    - {func_name: label_norm, xyxy2xywh_: True}
-    - {func_name: label_pad, padding_size: 160, padding_value: -1}
+    - {func_name: letterbox, scaleup: False, only_image: True}
     - {func_name: image_norm, scale: 255. }
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov5/hyp.scratch-low.yaml b/configs/yolov5/hyp.scratch-low.yaml
@@ -29,19 +29,18 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - {func_name: mosaic, prob: 1.0, mosaic9_prob: 0.0, translate: 0.1, scale: 0.5}
-    - {func_name: mixup, prob: 0.0, alpha: 32.0, beta: 32.0, needed_mosaic: True}
-    - {func_name: label_norm, xyxy2xywh_: True}
+    - { func_name: mosaic, prob: 1.0 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0 }
     - {func_name: albumentations}
     - {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - {func_name: fliplr, prob: 0.5}
+    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: label_pad, padding_size: 160, padding_value: -1}
     - {func_name: image_norm, scale: 255.}
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
 
   test_transforms:
-    - {func_name: letterbox, scaleup: False}
-    - {func_name: label_norm, xyxy2xywh_: True}
-    - {func_name: label_pad, padding_size: 160, padding_value: -1}
+    - {func_name: letterbox, scaleup: False, only_image: True}
     - {func_name: image_norm, scale: 255. }
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov7/hyp.scratch.p5.yaml b/configs/yolov7/hyp.scratch.p5.yaml
@@ -28,19 +28,23 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - {func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2, translate: 0.2, scale: 0.9}
-    - {func_name: mixup, prob: 0.15, alpha: 8.0, beta: 8.0, needed_mosaic: True}
+    - { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.2, scale: 0.9, shear: 0.0 }
+    - { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 0.15, pre_transform: [
+        { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.2, scale: 0.9, shear: 0.0 }, ]
+    }
     - {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - {func_name: pastein, prob: 0.15, num_sample: 30}
-    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: fliplr, prob: 0.5}
+    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: label_pad, padding_size: 160, padding_value: -1}
     - {func_name: image_norm, scale: 255.}
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
 
   test_transforms:
-    - {func_name: letterbox, scaleup: False}
-    - {func_name: label_norm, xyxy2xywh_: True}
-    - {func_name: label_pad, padding_size: 160, padding_value: -1}
+    - {func_name: letterbox, scaleup: False, only_image: True}
     - {func_name: image_norm, scale: 255. }
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov7/hyp.scratch.p6.yaml b/configs/yolov7/hyp.scratch.p6.yaml
@@ -28,19 +28,23 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - {func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2, translate: 0.2, scale: 0.9}
-    - {func_name: mixup, prob: 0.15, alpha: 8.0, beta: 8.0, needed_mosaic: True}
+    - { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.2, scale: 0.9, shear: 0.0 }
+    - { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 0.15, pre_transform: [
+        { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.2, scale: 0.9, shear: 0.0 }, ]
+    }
     - {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - {func_name: pastein, prob: 0.15, num_sample: 30}
-    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: fliplr, prob: 0.5}
+    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: label_pad, padding_size: 160, padding_value: -1}
     - {func_name: image_norm, scale: 255.}
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
 
   test_transforms:
-    - {func_name: letterbox, scaleup: False}
-    - {func_name: label_norm, xyxy2xywh_: True}
-    - {func_name: label_pad, padding_size: 160, padding_value: -1}
+    - {func_name: letterbox, scaleup: False, only_image: True}
     - {func_name: image_norm, scale: 255. }
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov7/hyp.scratch.tiny.yaml b/configs/yolov7/hyp.scratch.tiny.yaml
@@ -28,19 +28,23 @@ data:
   num_parallel_workers: 4
 
   train_transforms:
-    - {func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2, translate: 0.1, scale: 0.5}
-    - {func_name: mixup, prob: 0.05, alpha: 8.0, beta: 8.0, needed_mosaic: True}
+    - { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 }
+    - { func_name: resample_segments }
+    - { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0 }
+    - { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 0.05, pre_transform: [
+        { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.2 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0 }, ]
+    }
     - {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4}
     - {func_name: pastein, prob: 0.05, num_sample: 30}
-    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: fliplr, prob: 0.5}
+    - {func_name: label_norm, xyxy2xywh_: True}
     - {func_name: label_pad, padding_size: 160, padding_value: -1}
     - {func_name: image_norm, scale: 255.}
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
 
   test_transforms:
-    - {func_name: letterbox, scaleup: False}
-    - {func_name: label_norm, xyxy2xywh_: True}
-    - {func_name: label_pad, padding_size: 160, padding_value: -1}
+    - {func_name: letterbox, scaleup: False, only_image: True}
     - {func_name: image_norm, scale: 255. }
     - {func_name: image_transpose, bgr2rgb: True, hwc2chw: True }
diff --git a/configs/yolov8/hyp.scratch.high.yaml b/configs/yolov8/hyp.scratch.high.yaml
@@ -28,33 +28,40 @@ data:
     stage_epochs: [ 490, 10 ],
     trans_list: [
       [
-        {func_name: mosaic, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0, copy_paste_prob: 0.3},
-        {func_name: mixup, prob: 0.15, alpha: 32.0, beta: 32.0, needed_mosaic: True},
-        {func_name: label_norm, xyxy2xywh_: True},
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: copy_paste, prob: 0.3 },
+        {func_name: resample_segments},
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 },
+        { func_name: mixup, alpha: 32.0, beta: 32.0, prob: 0.15, pre_transform: [
+          { func_name: mosaic, prob: 1.0 },
+          { func_name: copy_paste, prob: 0.3 },
+          { func_name: resample_segments },
+          { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }, ]
+        },
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ],
       [
         {func_name: letterbox, scaleup: True},
+        {func_name: resample_segments},
         {func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0},
-        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ]]
   }
 
   test_transforms: [
-    {func_name: letterbox, scaleup: False},
-    {func_name: label_norm, xyxy2xywh_: True},
-    {func_name: label_pad, padding_size: 160, padding_value: -1},
+    {func_name: letterbox, scaleup: False, only_image: True},
     {func_name: image_norm, scale: 255.},
     {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
   ]
diff --git a/configs/yolov8/hyp.scratch.low.yaml b/configs/yolov8/hyp.scratch.low.yaml
@@ -30,33 +30,33 @@ data:
     stage_epochs: [ 490, 10 ],
     trans_list: [
       [
-        {func_name: mosaic, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0},
-        {func_name: mixup, prob: 0.0, alpha: 32.0, beta: 32.0, needed_mosaic: True},
-        {func_name: label_norm, xyxy2xywh_: True},
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0 },
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ],
       [
         {func_name: letterbox, scaleup: True},
+        {func_name: resample_segments},
         {func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.5, shear: 0.0},
-        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ]]
   }
 
   test_transforms: [
-    {func_name: letterbox, scaleup: False},
-    {func_name: label_norm, xyxy2xywh_: True},
-    {func_name: label_pad, padding_size: 160, padding_value: -1},
+    {func_name: letterbox, scaleup: False, only_image: True},
     {func_name: image_norm, scale: 255.},
     {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
   ]
diff --git a/configs/yolov8/hyp.scratch.med.yaml b/configs/yolov8/hyp.scratch.med.yaml
@@ -28,33 +28,40 @@ data:
     stage_epochs: [ 490, 10 ],
     trans_list: [
       [
-        {func_name: mosaic, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0, copy_paste_prob: 0.1},
-        {func_name: mixup, prob: 0.1, alpha: 32.0, beta: 32.0, needed_mosaic: True},
-        {func_name: label_norm, xyxy2xywh_: True},
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: copy_paste, prob: 0.1 },
+        {func_name: resample_segments},
+        { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 },
+        { func_name: mixup, alpha: 32.0, beta: 32.0, prob: 0.1, pre_transform: [
+          { func_name: mosaic, prob: 1.0 },
+          { func_name: copy_paste, prob: 0.1 },
+          { func_name: resample_segments },
+          { func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0 }, ]
+        },
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ],
       [
         {func_name: letterbox, scaleup: True},
+        {func_name: resample_segments},
         {func_name: random_perspective, prob: 1.0, degrees: 0.0, translate: 0.1, scale: 0.9, shear: 0.0},
-        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: albumentations},
         {func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4},
         {func_name: fliplr, prob: 0.5},
+        {func_name: label_norm, xyxy2xywh_: True},
         {func_name: label_pad, padding_size: 160, padding_value: -1},
         {func_name: image_norm, scale: 255.},
         {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
       ]]
   }
 
   test_transforms: [
-    {func_name: letterbox, scaleup: False},
-    {func_name: label_norm, xyxy2xywh_: True},
-    {func_name: label_pad, padding_size: 160, padding_value: -1},
+    {func_name: letterbox, scaleup: False, only_image: True},
     {func_name: image_norm, scale: 255.},
     {func_name: image_transpose, bgr2rgb: True, hwc2chw: True}
   ]
diff --git a/configs/yolox/hyp.scratch.yaml b/configs/yolox/hyp.scratch.yaml
@@ -39,20 +39,27 @@ data:
     stage_epochs: [ 285, 15 ],
     trans_list: [
       [
-        { func_name: mosaic, prob: 1.0, mosaic9_prob: 0.0, translate: 0.1, scale: 0.9, degrees: 10.0, shear: 2.0 },
-        { func_name: mixup, prob: 1.0, alpha: 8.0, beta: 8.0, needed_mosaic: True },
+        { func_name: mosaic, prob: 1.0 },
+        { func_name: resample_segments },
+        { func_name: random_perspective, prob: 1.0, degrees: 10.0, translate: 0.1, scale: 0.9, shear: 2.0 },
+        { func_name: mixup, alpha: 8.0, beta: 8.0, prob: 1.0, pre_transform: [
+          { func_name: mosaic, prob: 1.0 },
+          { func_name: resample_segments },
+          { func_name: random_perspective, prob: 1.0, degrees: 10.0, translate: 0.1, scale: 0.9, shear: 2.0 }, ]
+        },
         { func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 },
-        { func_name: label_norm, xyxy2xywh_: True },
         { func_name: fliplr, prob: 0.5 },
+        { func_name: label_norm, xyxy2xywh_: True },
         { func_name: label_pad, padding_size: 160, padding_value: -1 },
         { func_name: image_norm, scale: 255. },
         { func_name: image_transpose, bgr2rgb: True, hwc2chw: True },
       ],
       [
         { func_name: letterbox, scaleup: False },
+        { func_name: resample_segments },
         { func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 },
-        { func_name: label_norm, xyxy2xywh_: True },
         { func_name: fliplr, prob: 0.5 },
+        { func_name: label_norm, xyxy2xywh_: True },
         { func_name: label_pad, padding_size: 160, padding_value: -1 },
         { func_name: image_norm, scale: 255. },
         { func_name: image_transpose, bgr2rgb: True, hwc2chw: True },