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