mirror of
https://github.com/Deepshift/DeepCreamPy.git
synced 2024-12-01 04:00:21 +00:00
29 lines
1.3 KiB
Python
29 lines
1.3 KiB
Python
|
import argparse
|
||
|
|
|
||
|
|
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', dest='decensor_input_path', default='./decensor_input/', help='input images to be decensored by decensor.py path')
|
||
|
|
parser.add_argument('--decensor_output_path', dest='decensor_output_path', default='./decensor_output/', help='output images generated from running decensor.py path')
|
||
|
|
|
||
|
|
#Decensor settings
|
||
|
|
parser.add_argument('--mask_color_red', dest='mask_color_red', default=0, help='red channel of mask color in decensoring')
|
||
|
|
parser.add_argument('--mask_color_green', dest='mask_color_green', default=255, help='green channel of mask color in decensoring')
|
||
|
|
parser.add_argument('--mask_color_blue', dest='mask_color_blue', default=0, help='blue channel of mask color in decensoring')
|
||
|
|
|
||
|
|
parser.add_argument('--is_mosaic', dest='is_mosaic', default='False', type=str2bool, help='true if image has mosaic censoring, false otherwise')
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
return args
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
get_args()
|