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