Add more augmentaions

ngxbac · Jul 3, 2019 · af1d11b · af1d11b
1 parent 48038a4
commit af1d11b
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/bin/train.sh b/bin/train.sh
@@ -4,7 +4,7 @@ export CUDA_VISIBLE_DEVICES=2,3
 RUN_CONFIG=config.yml
 
 
-LOGDIR=/raid/bac/kaggle/logs/recursion_cell/test/rgb/se_resnext50_32x4d/
+LOGDIR=/raid/bac/kaggle/logs/recursion_cell/test/rgb_more_augs_2/se_resnext50_32x4d/
 catalyst-dl run \
     --config=./configs/${RUN_CONFIG} \
     --logdir=$LOGDIR \

diff --git a/src/augmentation.py b/src/augmentation.py
@@ -3,14 +3,19 @@
 
 def train_aug(image_size=224):
     return Compose([
-        RandomCrop(image_size, image_size),
-        HorizontalFlip(),
+        RandomCrop(448, 448),
+        Resize(image_size, image_size),
+        RandomRotate90(),
+        Flip(),
+        Transpose(),
+        # HorizontalFlip(),
         # Normalize(),
     ], p=1)
 
 
 def valid_aug(image_size=224):
     return Compose([
-        CenterCrop(image_size, image_size),
+        CenterCrop(448, 448),
+        Resize(320, 320)
         # Normalize(),
     ], p=1)
diff --git a/src/make_submission.py b/src/make_submission.py
@@ -34,7 +34,7 @@ def predict(model, loader):
 
 def predict_all():
     test_csv = '/raid/data/kaggle/recursion-cellular-image-classification/test.csv'
-    log_dir = "/raid/bac/kaggle/logs/recursion_cell/test/rgb/se_resnext50_32x4d/"
+    log_dir = "/raid/bac/kaggle/logs/recursion_cell/test/rgb_more_augs_2/se_resnext50_32x4d/"
     root = "/raid/data/kaggle/recursion-cellular-image-classification/"
     site = 1
     channels = [1,2,3]
@@ -74,7 +74,7 @@ def predict_all():
     submission = df.copy()
     submission['sirna'] = all_preds.astype(int)
     os.makedirs("submission", exist_ok=True)
-    submission.to_csv('./submission/submission_se_resnext50_32x4d_rgb.csv', index=False, columns=['id_code', 'sirna'])
+    submission.to_csv('./submission/submission_se_resnext50_32x4d_rgb_more_aug_2.csv', index=False, columns=['id_code', 'sirna'])
 
 
 if __name__ == '__main__':