DeepCreamPy/config.py

import argparse

def str2floatarr(v):
    if type(v) == str:
        try:
            return [float(v) for v in v.split(',')]
        except:
            raise argparse.ArgumentTypeError('Integers separated by commas expected.')
    else:
        raise argparse.ArgumentTypeError('Integers separated by commas expected.')

def str2bool(v):
    if v.lower() in ('yes', 'true', 't', 'y', '1', True):
        return True
    elif v.lower() in ('no', 'false', 'f', 'n', '0', False):
        return False
    else:
        raise argparse.ArgumentTypeError('Boolean value expected.')

def get_args():
	parser = argparse.ArgumentParser(description='')

	#Input output folders settings
	parser.add_argument('--decensor_input_path', default='./decensor_input/', help='input images with censored regions colored green to be decensored by decensor.py path')
	parser.add_argument('--decensor_input_original_path', default='./decensor_input_original/', help='input images with no modifications to be decensored by decensor.py path')
	parser.add_argument('--decensor_output_path', default='./decensor_output/', help='output images generated from running decensor.py path')

	#Decensor settings
	parser.add_argument('--mask_color_red', default=0, help='red channel of mask color in decensoring')
	parser.add_argument('--mask_color_green', default=255, help='green channel of mask color in decensoring')
	parser.add_argument('--mask_color_blue', default=0, help='blue channel of mask color in decensoring')
	parser.add_argument('--is_mosaic', type=str2bool, default='False', help='true if image has mosaic censoring, false otherwise')
	parser.add_argument('--variations', type=int, choices=[1, 2, 4], default=1, help='number of decensor variations to be generated')

	#Other settings
	parser.add_argument('--ui_mode', default=False, help='true if you want ui mode, false if you want command line interface')

	args = parser.parse_args()
	return args

if __name__ == '__main__':
	get_args()