Open 1ovebibi opened 2 years ago
One thing you could quickly try is changing the number of classes in the basemodel here: https://github.com/1ovebibi/Style-Transfer-for-Object-Detection-on-Historical-Artworks/blob/01deac4436d6113615a57d52e73f205931399b4b/configs/_base_/models/mask_rcnn_r50_fpn_unmasked.py#L47
And adapting the classes here: https://github.com/1ovebibi/Style-Transfer-for-Object-Detection-on-Historical-Artworks/blob/01deac4436d6113615a57d52e73f205931399b4b/configs/_base_/datasets/obj365_instance.py#L4
Although it should actually be overwritten by the inherited config file
Sorry I tried but it is not working
AssertionError: The
num_classes
(345) in Shared2FCBBoxHead of MMDataParallel does not matches the length ofCLASSES
80) in CocoDatasetThe training set patch0 has 345 classes in total. The parameter "num_classes" (in configs/swin/obj365-frcnn.py) was set to 345. All classes were written in the file configs/base/datasets/obj365_instence.py. All parameter info find in train-obj365.o503859