summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-18 18:34:24 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-18 18:36:47 +0800
commitf075fde91e6e5e7ac5d0f146df9cfde2b22fa150 (patch)
treef5c86dff6086ab8316a1f371585a0fab191aefa4 /config.py
parent31af04b70fcc0e7b46cbe18a52d150eb4e274f0e (diff)
parent84a3d5991f2f7272d1be54ad6cfe6ce695f915a0 (diff)
Merge branch 'master' into python3.8
Diffstat (limited to 'config.py')
-rw-r--r--config.py6
1 files changed, 3 insertions, 3 deletions
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,
},