utils: Change image size option

Author: synml

models/models.ini

@@ -4,7 +4,7 @@ activate_model = unet
 [unet]
 batch_size = 32
 epoch = 100
-image_size = 256
+image_size = 1024x512
 lr = 0.001
 num_classes = 8
 num_workers = 8
@@ -13,7 +13,7 @@ pretrained_weights = weights/unet_best.pth
 [proposed]
 batch_size = 10
 epoch = 100
-image_size = 256
+image_size = 1024x512
 lr = 0.001
 num_classes = 8
 num_workers = 8

utils/utils.py

@@ -18,7 +18,8 @@ def load_config():
     config = {
         'batch_size': parser.getint(model_name, 'batch_size'),
         'epoch': parser.getint(model_name, 'epoch'),
-        'image_size': parser.getint(model_name, 'image_size'),
+        'image_size': (int(parser[model_name]['image_size'].split('x')[0]),
+                       int(parser[model_name]['image_size'].split('x')[1])),
         'lr': parser.getfloat(model_name, 'lr'),
         'num_classes': parser.getint(model_name, 'num_classes'),
         'num_workers': parser.getint(model_name, 'num_workers'),