From 84a3d5991f2f7272d1be54ad6cfe6ce695f915a0 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 18 Feb 2021 18:34:06 +0800
Subject: Implement adjustable input size and change some default configs

---
 config.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'config.py')

diff --git a/config.py b/config.py
index 547d1a3..424bf5b 100644
--- a/config.py
+++ b/config.py
@@ -16,7 +16,7 @@ config: Configuration = {
         # Name of dataset (CASIA-B or FVG)
         'name': 'CASIA-B',
         # Path to dataset root (required)
-        'root_dir': 'data/CASIA-B-MRCNN/SEG',
+        'root_dir': 'data/CASIA-B-MRCNN-V2/SEG',
         # The number of subjects for training
         'train_size': 74,
         # Number of sampled frames per sequence (Training only)
@@ -27,8 +27,8 @@ config: Configuration = {
         'discard_threshold': 15,
         # Number of input channels of model
         'num_input_channels': 3,
-        # Resolution after resize, height : width should be 2 : 1
-        'frame_size': (64, 32),
+        # Resolution after resize, can be divided 16
+        'frame_size': (64, 48),
         # Cache dataset or not
         'cache_on': False,
     },
-- 
cgit v1.2.3