query2labels
query2labels copied to clipboard
fix:tiny bug in main_mlc.py
fix:tiny bug in main_mlc.py
line 405
'arch': args.arch,
should be
'arch': args.backbone,
or you should add another arch args
available_models = ['Q2L-R101-448', 'Q2L-R101-576', 'Q2L-TResL-448', 'Q2L-TResL_22k-448', 'Q2L-SwinL-384', 'Q2L-CvT_w24-384']
parser = argparse.ArgumentParser(description='Query2Label MSCOCO Training')
parser.add_argument('--dataname', help='dataname', default='coco14', choices=['coco14'])
parser.add_argument('--dataset_dir', help='dir of dataset', default='/comp_robot/liushilong/data/COCO14/')
parser.add_argument('--img_size', default=448, type=int,
help='size of input images')
parser.add_argument('-a', '--arch', metavar='ARCH', default='Q2L-R101-448',
choices=available_models,
help='model architecture: ' +
' | '.join(available_models) +
' (default: Q2L-R101-448)')
parser.add_argument('--output', metavar='DIR',
help='path to output folder')
parser.add_argument('--num_class', default=80, type=int,
help="Number of query slots")
parser.add_argument('--pretrained', dest='pretrained', action='store_true',
help='use pre-trained model. default is False. ')
parser.add_argument('--optim', default='AdamW', type=str, choices=['AdamW', 'Adam_twd'],
help='which optim to use')