yuhangzang / OV-DETR

[Under preparation] Code repo for "Open-Vocabulary DETR with Conditional Matching" (ECCV 2022)
202 stars 20 forks source link

How to train the model on customized dataset? #19

Open kinredon opened 1 year ago

kinredon commented 1 year ago

Hi, thank you for your excellent open-source work.

I am trying to train the model on my own dataset.

But I found the model needs to train a class-agnostic detector first and extract the features using the CLIP model.

However, no information is provided for the class-agnostic detector. Is there a convenient way for us to train the model on customized datasets?

I am looking forward to receiving your reply!