def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
load_size = 286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=13)
parser.set_defaults(contain_dontcare_label=False)
parser.add_argument(
'--label_dir',
type=str,
required=True,
help='path to the directory that contains label images')
parser.add_argument(
'--image_dir',
type=str,
required=True,
help='path to the directory that contains photo images')
parser.add_argument(
'--instance_dir',
type=str,
default='',
help=
'path to the directory that contains instance maps. Leave black if not exists'
)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if parser.get_default('use_512'):
load_size = 576 if is_train else 512 #286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
else:
load_size = 286 if is_train else 256 #286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=3) #13
parser.set_defaults(contain_dontcare_label=False)
# Resume
parser.add_argument('--data_root',
type=str,
default=r'/home/tangqf/SSD_disk/ITS/ITS')
parser.add_argument(
'--depth_dir',
type=str,
default=r'/home/Data/ZivonZhang/bts/Pro_fog',
help='path to the directory that contains depth images')
parser.add_argument('--train_file',
type=str,
default=r'train_file.txt')
parser.add_argument('--val_file', type=str, default=r'val_file.txt')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='none')
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=1)
parser.set_defaults(contain_dontcare_label=False)
parser.add_argument('--label_dir',
type=str,
required=True,
help='directory that contains segmentation masks')
parser.add_argument('--image_dir',
type=str,
required=True,
help='directory that contains images')
parser.add_argument('--condition_nc',
type=int,
default=3,
help='the number of lesion classes')
parser.add_argument('--isic_meta_data',
type=str,
default='./datasets/isic/meta_data.json',
help='path of meta data file')
parser.add_argument(
'--instance_dir',
type=str,
default='',
help=
'path to the directory that contains instance maps. Leave black if not exists'
)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=8)
opt, _ = parser.parse_known_args()
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='fixed')
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(aspect_ratio=1.0)
opt, _ = parser.parse_known_args()
if hasattr(opt, 'num_upsampling_layers'):
parser.set_defaults(num_upsampling_layers='more')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='crop')
parser.set_defaults(label_nc=1)
parser.set_defaults(no_label=True)
parser.set_defaults(no_pairing_check=True)
parser.set_defaults(no_instance=True)
parser.set_defaults(tf_log=True)
parser.add_argument('--instance_dir', type=str, default='',
help='path to the directory that contains instance maps. Leave black if not exists')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(dataroot='./dataset/facades/')
parser.set_defaults(preprocess_mode='resize_and_crop')
load_size = 286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=13)
parser.set_defaults(contain_dontcare_label=False)
parser.set_defaults(no_instance=True)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='fixed')
parser.set_defaults(load_size=512)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
parser.set_defaults(label_nc=35)
parser.set_defaults(aspect_ratio=2.0)
parser.set_defaults(dataroot='/home/tzt/dataset/cityscapes/')
opt, _ = parser.parse_known_args()
if hasattr(opt, 'num_upsampling_layers'):
parser.set_defaults(num_upsampling_layers='more')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='fixed_crop_resize')
parser.set_defaults(load_size=512)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
parser.set_defaults(label_nc=19)
parser.set_defaults(contain_dontcare_label=True)
parser.set_defaults(aspect_ratio=2.0)
opt, _ = parser.parse_known_args()
if hasattr(opt, 'num_upsampling_layers'):
parser.set_defaults(num_upsampling_layers='more')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='fixed')
parser.set_defaults(load_size=512)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
#need to change to 36 when we use background as another class
parser.set_defaults(label_nc=35)
parser.set_defaults(aspect_ratio=2.0)
opt, _ = parser.parse_known_args()
if hasattr(opt, 'num_upsampling_layers'):
parser.set_defaults(num_upsampling_layers='more')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=286)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=150)
parser.set_defaults(contain_dontcare_label=True)
parser.set_defaults(cache_filelist_read=False)
parser.set_defaults(cache_filelist_write=False)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.add_argument('--test_mode', type=str, default='all',
help='specify style mode to control multi-modal image synthesis (MMIS) during test phase:'
'night | cloudy | rainy | snowy | all')
parser.set_defaults(preprocess_mode='fixed')
parser.set_defaults(load_size=512)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
parser.set_defaults(aspect_ratio=2.0)
opt, _ = parser.parse_known_args()
if hasattr(opt, 'num_upsampling_layers'):
parser.set_defaults(num_upsampling_layers='more')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.add_argument('--coco_no_portraits', action='store_true')
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=286)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=182)
parser.set_defaults(contain_dontcare_label=True)
parser.set_defaults(cache_filelist_read=True)
parser.set_defaults(cache_filelist_write=True)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=286)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=150)
parser.set_defaults(contain_dontcare_label=True)
parser.set_defaults(cache_filelist_read=False)
parser.set_defaults(cache_filelist_write=False)
parser.set_defaults(no_instance=True)
parser.set_defaults(
dataroot=
'/home/tzt/HairSynthesis/SPADE/datasets/ADEChallengeData2016/')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=286)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=40)
parser.set_defaults(contain_dontcare_label=True)
parser.set_defaults(cache_filelist_read=False)
parser.set_defaults(cache_filelist_write=False)
parser.set_defaults(no_instance=True)
parser.add_argument('--train_list',
type=str,
help='import list of training folders')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode="resize_and_crop")
# parser.set_defaults(load_size=load_size)
# parser.set_defaults(crop_size=256)
# parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=2)
parser.set_defaults(contain_dontcare_label=False)
# parser.add_argument('--data_dir', type=str, default='/mnt/lvdisk1/tzt/HairSynthesis/SPADE-master/datasets/FFHQ',
# help='path to the directory that contains training & val data')
parser.add_argument(
"--label_dir",
type=str,
default="train_labels",
help="path to the directory that contains label images",
)
parser.add_argument(
"--image_dir",
type=str,
default="train_images",
help="path to the directory that contains photo images",
)
parser.add_argument(
"--instance_dir",
type=str,
default="",
help=
"path to the directory that contains instance maps. Leave black if not exists",
)
parser.add_argument(
"--orient_dir",
type=str,
default="train_dense_orients",
help="path to the directory that contains orientation mask",
)
parser.add_argument(
"--clear",
type=str,
default="",
help="[ |clear_], clear_ means use the selected training data",
)
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
load_size = 284 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
else:
load_size = 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=load_size)
parser.set_defaults(display_winsize=load_size)
parser.add_argument('--origin_dir', type=str, required=True)
parser.add_argument('--image_dir', type=str, default='')
parser.add_argument('--masks_dir', type=str, default='')
parser.add_argument('--mask_dir', type=str, default='')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='none')
parser.set_defaults(load_size=256)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=5)
parser.set_defaults(contain_dontcare_label=False)
parser.add_argument('--label_dir', type=str, required=True, help='directory that contains label images')
parser.add_argument('--image_dir_t1ce', type=str, required=True, help='directory that contains t1ce modality')
parser.add_argument('--image_dir_flair', type=str, required=True, help='directory that contains flair modality')
parser.add_argument('--image_dir_t2', type=str, required=True, help='directory that contains t2 modality')
parser.add_argument('--image_dir_t1', type=str, required=True, help='directory that contains t1 modality')
parser.add_argument('--condition_nc', type=int, default=10, help='the number of scanner classes')
parser.add_argument('--instance_dir', type=str, default='',
help='path to the directory that contains instance maps. Leave black if not exists')
return parser
def modify_commandline_options(parser, is_train):
parser = Pix2pixDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if parser.get_default('use_512'):
load_size = 576 if is_train else 512 #286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=512)
parser.set_defaults(display_winsize=512)
else:
load_size = 286 if is_train else 256 #286 if is_train else 256
parser.set_defaults(load_size=load_size)
parser.set_defaults(crop_size=256)
parser.set_defaults(display_winsize=256)
parser.set_defaults(label_nc=3) #13
parser.set_defaults(contain_dontcare_label=False)
parser.add_argument(
'--label_dir',
type=str,
default='/home/tangqf/SSD_disk/Fog_Data/NYU_Hazy',
help='path to the directory that contains label images')
parser.add_argument(
'--image_dir',
type=str,
default='/home/tangqf/SSD_disk/Fog_Data/NYU_GT',
help='path to the directory that contains photo images')
parser.add_argument(
'--depth_dir',
type=str,
default='/home/Data/ZivonZhang/bts/D_hazy',
help='path to the directory that contains depth images')
parser.add_argument(
'--instance_dir',
type=str,
default='',
help=
'path to the directory that contains instance maps. Leave black if not exists'
)
return parser