用自己数据集训练lw-detr
要在相应位置修改自己数据集的类数。
LW_DETR/models/lwdetr.py
中第571行的num_classes = 20 if args.dataset_file != 'coco' else 91
修改为:num_classes = 20 if args.dataset_file != 'coco' else 102
要在相应位置修改自己数据集的类数。
LW_DETR/models/lwdetr.py
中第571行的num_classes = 20 if args.dataset_file != 'coco' else 91
修改为:num_classes = 20 if args.dataset_file != 'coco' else 102