update

segtrain.py

@@ -107,7 +107,7 @@ def get_training_augmentation():
 
 def get_validation_augmentation():
     test_transform = [
-        albu.PadIfNeeded(384, 480)
+        albu.PadIfNeeded(512, 512)
     ]
     return albu.Compose(test_transform)
 
@@ -142,8 +142,8 @@ if __name__ == '__main__':
     y_valid_dir = os.path.join(DATA_DIR, 'valannot')
 
     #ENCODER = 'se_resnext50_32x4d'
-    #ENCODER = 'resnet18'
-    ENCODER = 'mobilenet_v2'
+    ENCODER = 'resnet18'
+    #ENCODER = 'mobilenet_v2'
     ENCODER_WEIGHTS = 'imagenet'
     CLASSES = ['front', 'background']
     ACTIVATION = 'sigmoid' # could be None for logits or 'softmax2d' for multiclass segmentation