用自己数据集训练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

posted @ 2025-03-20 21:04  anti1hapi  阅读(101)  评论(0)    收藏  举报