summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-18 18:38:31 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-18 18:38:31 +0800
commit8012fce3e595aad31f4a52dc316b46e558234dff (patch)
tree4a5e19a21ad8a4470931f5884777c127197153c0 /config.py
parent2988c1b9afd4e869b629a8629abedbf63d2452aa (diff)
parent84a3d5991f2f7272d1be54ad6cfe6ce695f915a0 (diff)
Merge branch 'master' into data_parallel_py3.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,
},