search

PeizeSun / SparseR-CNN

[CVPR2021, PAMI2023] End-to-End Object Detection with Learnable Proposal
MIT License
1.31k stars 187 forks source link

How can I use a custom dataset? #99

Open JEO-96 opened 2 years ago

JEO-96 commented 2 years ago

I want to use a custom dataset other than the coco dataset. How do I change the path?

PeizeSun commented 2 years ago

Hi~ Our code is based on Detectron2, the way of using custom dataset is the same as detectron2, I guess you could refer to Use Custom Dataset.

JEO-96 commented 2 years ago

Thank you for your answer. But not all problems have been solved. Through the detectron2 tutorial, I learned how to use customized datasets. However, I couldn't figure out how to use the model you created while using the custom data set. I can only use the model supported by detectron 2 through the code below. cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcn_R_50_FPN_3x.yaml") How can I use your model? Also, what I'm worried about is that I didn't use the head you made because I followed the detectron 2 tutorial.