def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.add_argument(
'--n_frames_total',
type=int,
default=30,
help='the overall number of frames in a sequence to train with')
parser.add_argument(
'--max_frames_per_gpu',
type=int,
default=6,
help='number of frames to load into one GPU at a time')
parser.add_argument(
'--n_frames_pre_load_test',
type=int,
default=1,
help='number of frames to load every time when test')
parser.add_argument(
'--total_test_frames',
type=int,
default=300,
help='the overall number of frames to load every sequence when test'
)
parser.add_argument(
'--max_t_step',
type=int,
default=1,
help=
'max spacing between neighboring sampled frames. If greater than 1, the network may randomly skip frames during training.'
)
parser.add_argument('--debug', action='store_true', help='debuge mode')
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='center_crop_and_resize')
parser.set_defaults(center_crop_size=178)
parser.set_defaults(load_size=128)
parser.set_defaults(crop_size=128)
parser.set_defaults(start_size=16)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
assert isinstance(parser, argparse.ArgumentParser)
parser.add_argument('--meta_path',
type=str,
default=None,
help='the path to the meta file')
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
if is_train:
parser.set_defaults(load_size=256)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(old_size=(256, 256))
parser.set_defaults(structure_nc=18)
parser.set_defaults(image_nc=3)
parser.set_defaults(display_winsize=256)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.add_argument('--ang_skip', type=int, default=20)
parser.add_argument('--label_nc_h', type=int, default=18)
parser.add_argument('--label_nc_v', type=int, default=3)
parser.add_argument('--sub_dataset_model', type=str, default='car')
parser.set_defaults(load_size=256)
parser.set_defaults(image_nc=3)
parser.set_defaults(display_winsize=256)
parser.set_defaults(structure_nc=18 + 3)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=256)
else:
parser.set_defaults(load_size=256)
parser.set_defaults(fid_gt_path='./dataset/fashion_data/train_256')
parser.set_defaults(old_size=(256, 176))
parser.set_defaults(structure_nc=18)
parser.set_defaults(image_nc=3)
parser.set_defaults(display_winsize=256)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(model='indoor')
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=29)
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(no_pairing_check=True)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
if is_train:
parser.set_defaults(load_size=(128, 64))
parser.set_defaults(angle=(-5, 5))
parser.set_defaults(shift=(-5, 5))
parser.set_defaults(scale=(0.95, 1.05))
else:
parser.set_defaults(load_size=(128, 64))
parser.set_defaults(angle=False)
parser.set_defaults(shift=False)
parser.set_defaults(scale=False)
parser.set_defaults(old_size=(128, 64))
parser.set_defaults(structure_nc=18)
parser.set_defaults(image_nc=3)
parser.set_defaults(display_winsize=128)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.set_defaults(preprocess_mode='resize_and_crop')
if is_train:
parser.set_defaults(load_size=(128, 64))
parser.set_defaults(angle=(-5, 5))
parser.set_defaults(shift=(-5, 5))
parser.set_defaults(scale=(0.95, 1.05))
else:
parser.set_defaults(load_size=(128, 64))
parser.set_defaults(angle=False)
parser.set_defaults(shift=False)
parser.set_defaults(scale=False)
parser.set_defaults(fid_gt_path='./dataset/market_data/train_128')
parser.set_defaults(old_size=(128, 64))
parser.set_defaults(structure_nc=18)
parser.set_defaults(image_nc=3)
parser.set_defaults(display_winsize=128)
return parser
def modify_commandline_options(parser, is_train):
parser = BaseDataset.modify_commandline_options(parser, is_train)
parser.add_argument(
'--t_step',
type=int,
default=1,
help=
'max spacing between neighboring sampled frames. If greater than 1, the network may randomly skip frames during training.'
)
parser.add_argument('--n_frames_pre_load',
type=int,
default=24,
help='number of output keypoints pre load')
parser.add_argument('--n_frames_pre_load_test',
type=int,
default=24,
help='number of output keypoints pre load')
parser.add_argument('--n_receptive',
type=int,
default=81,
help='The receptive fields of the network')
parser.add_argument('--gt_path',
type=str,
default='./dataset/human36m/data_2d_h36m_gt.npz')
parser.add_argument(
'--input_path',
type=str,
default='./dataset/human36m/data_2d_h36m_detectron_pt_coco.npz')
parser.add_argument('--debug', action='store_true', help='debuge mode')
parser.set_defaults(debug=False)
parser.set_defaults(structure_nc=17)
parser.set_defaults(batchSize=32)
if not is_train:
parser.set_defaults(batchSize=1)
return parser
