semantic-segmentation/initialize.py at master · marcelsan/semantic-segmentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""
	This file defines all the parameters for the experiment.
	Author: Marcel Santos (mss8@cin.ufpe.br), Federal University of Pernambuco (UFPE).
"""

import tensorflow as tf

FLAGS = tf.app.flags.FLAGS

tf.app.flags.DEFINE_string  ('experimentName', 'seg_net_skip_concatenate_adam_lr_01_block_4',
                            'Name of the experiment.')

tf.app.flags.DEFINE_boolean ('saveModel', 1,
                            'Indicate whether the model json must be saved.')

tf.app.flags.DEFINE_boolean ('saveFinalWeights', 1,
                            'Indicate whether the final weights of the model must be saved.')

tf.app.flags.DEFINE_integer ('numClasses', 21,
							'Indicate the number of classes on the dataset.')

tf.app.flags.DEFINE_string  ('trainImageDir', 'dataset/train/input/',
                            'Directory for the train set input.')

tf.app.flags.DEFINE_string  ('trainLabelsDir', 'dataset/train/segments/',
                            'Directory for the train set labels.')

tf.app.flags.DEFINE_string  ('valImagesDir', 'dataset/val/input/',
                            'Directory for the validation set input.')

tf.app.flags.DEFINE_string  ('valLabelsDir', 'dataset/val/segments/',
                            'Directory for the validation set labels.')