add 2 more data augmentation

wangxdflight · Oct 30, 2022 · d0e7f20 · d0e7f20
1 parent 42d1f6b
commit d0e7f20
Show file tree

Hide file tree

Showing 7 changed files with 36 additions and 30 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,4 +5,5 @@ __pycache__
 models/*
 .idea/*
 *.bak
-pretrained/*
+pretrained/*
+.idea/workspace.xml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/configs_tf2/ssd_mobilenet_v2_320x320_coco17_tpu-8.config b/configs_tf2/ssd_mobilenet_v2_320x320_coco17_tpu-8.config
@@ -157,6 +157,14 @@ train_config: {
     ssd_random_crop {
     }
   }
+  data_augmentation_options {
+    random_adjust_contrast {
+    }
+  }
+  data_augmentation_options {
+    random_patch_gaussian {
+    }
+  }
   optimizer {
     momentum_optimizer: {
       learning_rate: {

diff --git a/configs_tf2/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.config b/configs_tf2/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.config
@@ -160,6 +160,14 @@ train_config: {
       overlap_thresh: 0.0
     }
   }
+  data_augmentation_options {
+    random_adjust_contrast {
+    }
+  }
+  data_augmentation_options {
+    random_patch_gaussian {
+    }
+  }
   optimizer {
     momentum_optimizer: {
       learning_rate: {

diff --git a/data/ssd_mobilenet_v2_320x320_coco17_tpu-8.config b/data/ssd_mobilenet_v2_320x320_coco17_tpu-8.config
@@ -157,6 +157,14 @@ train_config: {
     ssd_random_crop {
     }
   }
+  data_augmentation_options {
+    random_adjust_contrast {
+    }
+  }
+  data_augmentation_options {
+    random_patch_gaussian {
+    }
+  }
   optimizer {
     momentum_optimizer: {
       learning_rate: {

diff --git a/data/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.config b/data/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8.config
@@ -160,6 +160,14 @@ train_config: {
       overlap_thresh: 0.0
     }
   }
+  data_augmentation_options {
+    random_adjust_contrast {
+    }
+  }
+  data_augmentation_options {
+    random_patch_gaussian {
+    }
+  }
   optimizer {
     momentum_optimizer: {
       learning_rate: {

diff --git a/resize_images.py b/resize_images.py
@@ -2,6 +2,8 @@
 import glob
 import cv2
 
+# python resize_images.py --raw-dir ./data/raw --save-dir ./data/images --ext jpg --target-size "(800, 600)"
+
 if __name__ == "__main__":
     import argparse