import tensorflow as tf

# model related configuration
tf.app.flags.DEFINE_integer(
    'train_image_size', 352,
    'The size of the input image for the model to use.')
tf.app.flags.DEFINE_integer(
    'resnet_size', 50,
    'The size of the ResNet model to use.')
tf.app.flags.DEFINE_integer(
