Object detection on custom datasets?

[sedol1339]

Hello! How can I fine-tune this model on a custom object detection dataset, that is already registered in DatasetCatalog?

In MaskDINO/configs folder I see only configs for segmentation, is there any configs for detection?

[RYHSmmc]

Same questions.

[FengLi-ust]

By default, we train detection and segmentation at the same time. If you want to only train detection, you need to modify the prediction head to not predict mask and not calculate mask loss. You can directly use DINO if you only do detection.

We can update the code if you need this function.

[anirudh6415]

@FengLi-ust can you please update the code, so that it can only do detection if required and segmentation separately.