def parse_args(): parser = argparse.ArgumentParser(description='Train keypoints network') parser.add_argument( '--cfg', help='experiment configure file name', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument( '--frequent', help='frequency of logging', default=config.PRINT_FREQ, type=int) parser.add_argument('--gpus', help='gpus', type=str) parser.add_argument('--workers', help='num of dataloader workers', type=int) parser.add_argument( '--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument( '--dataDir', help='data directory', type=str, default='') parser.add_argument( '--data-format', help='data format', type=str, default='') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Train keypoints network') parser.add_argument( '--cfg', help='experiment configure file name', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument( '--frequent', help='frequency of logging', default=config.PRINT_FREQ, type=int) parser.add_argument('--gpus', help='gpus', type=str) parser.add_argument('--workers', help='num of dataloader workers', type=int) parser.add_argument( '--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument( '--dataDir', help='data directory', type=str, default='') parser.add_argument( '--data-format', help='data format', type=str, default='') parser.add_argument( '--on-server-cluster', help='if training on cluster', action='store_true') parser.add_argument( '--iteration', help='the kth times of training', type=int, choices=range(1,10), default=1) parser.add_argument( '--no-distortion', help='wheter use no distortion data', action='store_true') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Evaluate 3D Pose Estimation') parser.add_argument('--cfg', help='configuration file name', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument('--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument('--dataDir', help='data directory', type=str, default='') parser.add_argument('--withIMU', help='use bone orientation in 3d', type=lambda x: (str(x).lower() in ['true', '1', 'yes']), default=False) args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Generate Data For RPSM') parser.add_argument('--cfg', help='configuration file name', required=True, type=str) parser.add_argument('--heatmap', help='heatmap file name', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument('--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument('--dataDir', help='data directory', type=str, default='') parser.add_argument('--no-distortion', help='wheter use no distortion data', action='store_true') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Train keypoints network') # general parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str) args, rest = parser.parse_known_args() # update config update_config(args.cfg) # training parser.add_argument('--frequent', help='frequency of logging', default=config.PRINT_FREQ, type=int) parser.add_argument('--gpus', help='gpus', type=str) parser.add_argument( '--state', help='the state of model which is used to test (best or final)', default='best', type=str) parser.add_argument('--workers', help='num of dataloader workers', type=int) parser.add_argument('--model-file', help='model state file', type=str) parser.add_argument('--flip-test', help='use flip test', action='store_true') parser.add_argument('--post-process', help='use post process', action='store_true') parser.add_argument('--shift-heatmap', help='shift heatmap', action='store_true') # philly parser.add_argument('--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument('--dataDir', help='data directory', type=str, default='') parser.add_argument('--data-format', help='data format', type=str, default='') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Test Pseudo Labels') # Required Param parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str) parser.add_argument('--heatmap', help='heatmap file name', default='', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument('--confidence-thre', help='max pixel error to accept inliers', type=float) parser.add_argument('--ransac', help='use ransac', action='store_true') parser.add_argument('--inliers', help='min support num of inliers', type=int) parser.add_argument('--reproj-thre', help='reprojection threshold to accept inliers', type=int) parser.add_argument('--use-reproj', help='use reprojected 2d as label', action='store_true') parser.add_argument( '--loop', help='whether in loop training, only generate one pseudo_label', action='store_true') parser.add_argument('--no-distortion', help='whether use no distortion data', action='store_true') parser.add_argument('--net-layers', help='network layers', type=int, default=50, choices=[50, 152]) parser.add_argument('--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument('--dataDir', help='data directory', type=str, default='') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser(description='Train keypoints network') parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument('--frequent', help='frequency of logging', default=config.PRINT_FREQ, type=int) parser.add_argument('--gpus', help='gpus', type=str) parser.add_argument('--workers', help='num of dataloader workers', type=int) parser.add_argument('--cams', help='view/cam id to use', type=str) # parser.add_argument( # '--ablation', help='heatmap_only, consistency_only, both', type=str) parser.add_argument('--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument('--dataDir', help='data directory', type=str, default='') parser.add_argument('--data-format', help='data format', type=str, default='') parser.add_argument('--runMode', help='train or test', type=str, default='train') parser.add_argument('--modelFile', help='model for test', type=str, default='0') parser.add_argument('--evaluate', help='directly use provided model to evaluate results', type=str2bool, default='0') args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args
def parse_args(): parser = argparse.ArgumentParser( description='Test Recursive Pictorial Structure Model') parser.add_argument( '--cfg', help='experiment configure file name', required=True, type=str) parser.add_argument( '--heatmap', help='heatmap file name', default='', type=str) args, rest = parser.parse_known_args() update_config(args.cfg) parser.add_argument( '--modelDir', help='model directory', type=str, default='') parser.add_argument('--logDir', help='log directory', type=str, default='') parser.add_argument( '--dataDir', help='data directory', type=str, default='') parser.add_argument( '--no-distortion', help='wheter use no distortion data', action='store_true') parser.add_argument( '--inliers', help='min support num of inliers', type=int) parser.add_argument( '--reproj-thre', help='reprojection threshold to accept inliers', type=int) args = parser.parse_args() update_dir(args.modelDir, args.logDir, args.dataDir) return args