Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard

Model on Huggingface

Open daniel-bogdoll opened this issue 1 year ago • 1 comments

Hey,

thanks for the great work! Would you be interested in adding the necessary adjustments/configs to HuggingFace, so that the model can be loaded with the Auto functions from Huggingface transformers? Currently, https://huggingface.co/zongzhuofan/co-detr-vit-large-coco only provides a pth file.

from transformers import AutoImageProcessor, AutoModelForObjectDetection

image_processor = AutoImageProcessor.from_pretrained("zongzhuofan/co-detr-vit-large-coco")
model = AutoModelForObjectDetection.from_pretrained("zongzhuofan/co-detr-vit-large-coco")

One example where this was done is "facebook/detr-resnet-50".

daniel-bogdoll avatar Oct 17 '24 15:10 daniel-bogdoll

Thanks for your suggestion. However, my focus is currently on other projects, and I don't have much time to develop new features for this project right now. Community PR is welcome.

TempleX98 avatar Oct 25 '24 09:10 TempleX98