return args
if __name__ == "__main__":
#Parse commandline arguments
args = parse_args()
#Required params
input_params = InputParameters(args)
image_generation_params = ImageGenerationParameters(args)
num_prediction_steps = args.num_prediction_steps
dropbox_parameters = args.dropbox_parameters
log_to_console = args.log_to_console
#Initialize logging
logging.initialize(__file__, log_to_console = log_to_console)
logger = logging.get_logger(__name__)
#Log input parameters
logger.info('Running with parameters input_params: %s', input_params)
logger.info('Additional parameters: image_generation_params: %s log_to_console: %s', image_generation_params, log_to_console)
#Predictable randomness
seed = 3
np_seed(seed)
tf_seed(seed)
#Dropbox
dropbox = None
if dropbox_parameters:
def __init__(self, *args, **kwargs):
super(TestDropboxConnection, self).__init__(*args, **kwargs)
#Initialize logging
logging.initialize('unittest')
def __init__(self, methodName = 'runTest'):
super(TestImageDataTransformation, self).__init__(methodName)
#Logging
logging.initialize(ut_constants.LOGGING_CLASS)
self._logger = logging.get_logger(__name__)
args = parse_args()
dataset_location = args.dataset_location
output_dataset_location = args.output_dataset_location
input_file = args.input_file
output_file = args.output_file
num_inputs = args.num_inputs
image_col = args.image_col
target_shape = tuple(args.target_shape)
dropbox_parameters = args.dropbox_parameters
parallel = args.parallel
log_to_console = args.log_to_console
#Initialize logging
logging.initialize(__file__,
log_to_console=log_to_console,
no_logging=parallel)
logger = logging.get_logger(__name__)
#Log input parameters
logger.info(
'Parameters:: dataset_location: %s output_dataset_location: %s',
dataset_location, output_dataset_location)
logger.info('Parameters:: input_file: %s output_file: %s', input_file,
output_file)
logger.info('Parameters:: image_col: %s target_shape: %s', image_col,
target_shape)
logger.info('Parameters:: num_inputs: %s parallel: %s log_to_console: %s',
num_inputs, parallel, log_to_console)
####################################### Validation [Start] ############################################