search

mcahny / object_localization_network

Learning Open-World Object Proposals without Learning to Classify
Apache License 2.0
193 stars 26 forks source link

Number of class in training #10

Open mfanparc opened 2 years ago

mfanparc commented 2 years ago

Hi, thanks for the inspiring work. When I retrained the model with the config /oln_box, i received this: "AssertionError: The num_classes (1) in Shared2FCBBoxScoreHead of MMDataParallel does not matches the length of CLASSES 80) in CocoSplitDataset " So, should I set the num_classes = 80 instead of 1 when training with COCO? Or there is something else I should do. Thanks!