Co-DETR
Co-DETR copied to clipboard
Model on Huggingface
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".
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.