Commit 5b09c379 authored by Fernando Perez-Garcia's avatar Fernando Perez-Garcia

Modify some default parameters

parent 6c0588a7
......@@ -55,7 +55,8 @@ SEGMENTATION_NIFTYNET = '.'.join([
'segmentation_application',
'SegmentationApplication'
])
SEGMENTATION_APP = 'segmentation_application_mine.SegmentationApplicationMine'
SEGMENTATION_APP = 'vesseg_applications.segmentation_application_vessels.SegmentationApplicationVessels'
# Random seed
......@@ -80,9 +81,9 @@ VOXEL_SPACING = 0.466 # most DSAs
NEAREST_NEIGHBOR = 0
LINEAR = 1 # true?
X_INTERPOLATION = 3
TRAINING_IMAGE_WINDOW_SIZE = 64
TRAINING_IMAGE_WINDOW_SIZE = 96
TRAINING_LABEL_WINDOW_SIZE = TRAINING_IMAGE_WINDOW_SIZE
INFERENCE_WINDOW_SIZE = 64
INFERENCE_WINDOW_SIZE = 96
# System
NUM_THREADS = 2
......@@ -90,14 +91,14 @@ NUM_THREADS = 2
# Network
# NETWORK_NAME = UNET_SMALL
NETWORK_NAME = HIGHRES3DNET_SMALLER
BATCH_SIZE = 1
BATCH_SIZE = 4
WINDOW_SAMPLING = 'weighted'
VOLUME_PADDING_SIZE = TRAINING_IMAGE_WINDOW_SIZE // 2
# Training
MAX_NUM_ITERATIONS = 10000
SAVE_EVERY_N = MAX_NUM_ITERATIONS // 10
LEARNING_RATE = 1e-3
LEARNING_RATE = 1e-1
TENSORBOARD_EVERY_N = 1
VALIDATION_EVERY_N = 5
LATEST_CHECKPOINT = -1
......@@ -183,6 +184,8 @@ class Model:
self.config = ConfigParser()
self.application = SEGMENTATION_APP
def __repr__(self):
return self.name
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment