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
