python_window_data_rots.py 文件源码

def parse_args(cls, argsStr):
        parser = argparse.ArgumentParser(description='PythonWindowDataRots Layer')
        parser.add_argument('--source', default='', type=str)
        parser.add_argument('--root_folder', default='', type=str)
        parser.add_argument('--mean_file', default='', type=str)
        parser.add_argument('--batch_size', default=128, type=int)
        parser.add_argument('--crop_size', default=192, type=int)
        parser.add_argument('--is_gray', dest='is_gray', action='store_true')
        parser.add_argument('--no-is_gray', dest='is_gray', action='store_false')
        parser.add_argument('--is_random_roll',    dest='is_gray', action='store_true', default=True)
        parser.add_argument('--no-is_random_roll', dest='is_gray', action='store_false')
        parser.add_argument('--is_mirror',  dest='is_mirror', action='store_true', default=False)
        parser.add_argument('--resume_iter', default=0, type=int)
        parser.add_argument('--jitter_pct', default=0, type=float)
        parser.add_argument('--jitter_amt', default=0, type=int)
        parser.add_argument('--nrmlz_file', default='None', type=str)
        parser.add_argument('--ncpu', default=2, type=int)
        args   = parser.parse_args(argsStr.split())
        print('Using Config:')
        pprint.pprint(args)
        return args