how to extract features on my own dataset

[Chunmian-art]

Hi, thank you for the nice work! But I still have some questions on extract features on my own dataset. I train an object detection model through detectron2. But how can I use this model in your work? Or should I use your work train an object detection model?

[Zoroaster97]

This framework is not support training models yet (we are still working on it 😢). We provide some pre-trained models, you can use these models to extract features on your datasets. If you want to use your model, you can try to modify the "model" part of this framework, and replace it with your code.

[leonodelee]

This framework is not support training models yet (we are still working on it 😢). We provide some pre-trained models, you can use these models to extract features on your datasets. If you want to use your model, you can try to modify the "model" part of this framework, and replace it with your code.

I wonder if I can use my own model.pth to extract features on my datasets using your codebase?