Are object detection models supported in optimum?

🤗Optimum
1

are object detection models supported in optimum? I don’t see any tasks defined for OD here optimum/optimum/onnxruntime/trainer.py at main · huggingface/optimum · GitHub

2

Hi @sak99! What is your use case (training, inference, ONNX, BetterTransformer,…)? For instance it is possible to export models like DETR to the ONNX format.

3

Hi @regisss I am looking at Faster R-CNN, YOLO, SparseRCNN

4

DETR and YOLOS are the only object-detection models that can be exported to ONNX with Optimum. The models you mentioned are not supported by the Transformers library so they are not by Optimum either.