mirror of
https://github.com/Deepshift/DeepCreamPy.git
synced 2024-11-28 11:09:42 +00:00
62 lines
3.7 KiB
Python
62 lines
3.7 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.')
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='')
|
|
|
|
#Image settings
|
|
parser.add_argument('--input_size', dest='input_size', default=128, help='input image size')
|
|
parser.add_argument('--local_input_size', dest='local_input_size', default=64, help='local input image size')
|
|
parser.add_argument('--input_channel_size', dest='input_channel_size', default=3, help='input image channel')
|
|
parser.add_argument('--min_mask_size', dest='min_mask_size', default=24, help='minimum mask size')
|
|
parser.add_argument('--max_mask_size', dest='max_mask_size', default=48, help='maximum mask size')
|
|
parser.add_argument('--rotate_chance', dest='rotate_chance', default=0.7, help='chance the mask will be randomly rotated')
|
|
parser.add_argument('--train_mosaic', dest ='train_mosaic', default=False, help='train neural network to decensor mosaics')
|
|
|
|
# parser.add_argument('--input_dim', dest='input_dim', default=100, help='input z size')
|
|
|
|
# #Training settings
|
|
parser.add_argument('--continue_training', dest='continue_training', default=False, type=str2bool, help='flag to continue training')
|
|
parser.add_argument('--training_samples_path', dest='training_samples_path', default='./training_samples/', help='samples images generated during training path')
|
|
parser.add_argument('--batch_size', dest='batch_size', default=16, help='batch size')
|
|
|
|
# parser.add_argument('--data', dest='data', default='../ambientGAN_TF/data', help='cats image train path')
|
|
|
|
# parser.add_argument('--train_step', dest='train_step', default=400, help='total number of train_step')
|
|
# parser.add_argument('--Tc', dest='Tc', default=100, help='Tc to train Completion Network')
|
|
# parser.add_argument('--Td', dest='Td', default=1, help='Td to train Discriminator Network')
|
|
|
|
|
|
parser.add_argument('--learning_rate', dest='learning_rate', default=0.001, help='learning rate of the optimizer')
|
|
# parser.add_argument('--momentum', dest='momentum', default=0.5, help='momentum of the optimizer')
|
|
|
|
# #I set alpha to 1 to give more weights to the discriminator loss
|
|
# parser.add_argument('--alpha', dest='alpha', default=1.0, help='alpha')
|
|
|
|
# parser.add_argument('--margin', dest='margin', default=5, help='margin')
|
|
|
|
# #Test image
|
|
# parser.add_argument('--img_path', dest='img_path', default='', help='test image path')
|
|
|
|
# #Extra folders settings
|
|
# parser.add_argument('--checkpoints_path', dest='checkpoints_path', default='./checkpoints/', help='saved model checkpoint path')
|
|
# parser.add_argument('--graph_path', dest='graph_path', default='./graphs/', help='tensorboard graph')
|
|
# parser.add_argument('--images_path', dest='images_path', default='./images/', help='result images path')
|
|
parser.add_argument('--testing_output_path', dest='testing_output_path', default='./testing_output/', help='output images generated from running test.py path')
|
|
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')
|
|
|
|
args = parser.parse_args() |