def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', '-d', type=str, dest='gpu_ids')
parser.add_argument('--continue',
'-c',
dest='continue_train',
action='store_true')
parser.add_argument('--debug', dest='debug', action='store_true')
args = parser.parse_args()
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = 0 if not gpus[0].isdigit() else int(gpus[0])
gpus[1] = len(
mem_info()) if not gpus[1].isdigit() else int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', '-d', type=str, dest='gpu_ids')
parser.add_argument('--range', '-r', type=str, dest='test_epochs')
parser.add_argument('--model', '-m', type=str, dest='test_model')
args = parser.parse_args()
# test gpus
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = 0 if not gpus[0].isdigit() else int(gpus[0])
gpus[1] = len(mem_info()) if not gpus[1].isdigit() else int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
if args.test_epochs and not ',' in args.test_epochs:
args.test_epochs = '%d,%d' % (int(args.test_epochs), int(args.test_epochs) + 1)
assert args.test_epochs or args.test_model, 'Test epoch or model to test is required.'
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', type=str, dest='gpu_ids')
parser.add_argument('--test_epoch', type=str, dest='test_epoch')
parser.add_argument('--visatt', dest='visatt', action='store_true')
parser.add_argument('--imgid', type=str, dest='imgid')
parser.add_argument('--newtestdata', type=str, dest='newtestdata')
args = parser.parse_args()
# test gpus
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = 0 if not gpus[0].isdigit() else int(gpus[0])
gpus[1] = len(
mem_info()) if not gpus[1].isdigit() else int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
assert args.test_epoch, 'Test epoch is required.'
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', type=str, dest='gpu_ids')
parser.add_argument('--test_epoch', type=str, dest='test_epoch')
parser.add_argument('--vis', action='store_true', default=False)
parser.add_argument('--vis_occ', action='store_true', default=False)
parser.add_argument('--dataset', type=str, dest="dataset", default="")
parser.add_argument('--cfg', type=str, dest="cfg")
parser.add_argument('--use_dets', action='store_true', default=False)
args = parser.parse_args()
# test gpus
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = 0 if not gpus[0].isdigit() else int(gpus[0])
gpus[1] = len(
mem_info()) if not gpus[1].isdigit() else int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
assert args.test_epoch, 'Test epoch is required.'
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', type=str, dest='gpu_ids')
parser.add_argument('--test_epoch',
type=str,
dest='test_epoch',
default="last_epoch")
parser.add_argument('--image_file', type=str, help="Image file")
args = parser.parse_args()
# test gpus
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = 0 if not gpus[0].isdigit() else int(gpus[0])
gpus[1] = len(
mem_info()) if not gpus[1].isdigit() else int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
assert args.test_epoch, 'Test epoch is required.'
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', type=str, dest='gpu_ids')
parser.add_argument('--cfg',
type=str,
dest='cfg',
default='hrnet_w32_bs128_256x192_epoch210')
parser.add_argument('--continue',
dest='continue_train',
action='store_true')
args = parser.parse_args()
if not args.gpu_ids:
args.gpu_ids = str(np.argmin(mem_info()))
return args
