HumanSignal / HumanSignal/label-studio-ml-backend

Question: Integrating a Fine-tuned DETR Model for Object Detection

Open
#800 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.1k
Forks
490
Avg merge
1d 14h
Merged PRs (30d)
3

Description

Hi HumanSignal team👋
First and foremost, thank you for all your hard work on Label Studio and its ML backend. It's a great tool for the data science community, and I'm very grateful for the effort you've put into open-sourcing it. 😻🌟

I have a question regarding the integration of a custom model:

I've fine-tuned a DETR-based model for a specific object detection use case and would like to integrate it into the Label Studio ML backend. I've successfully run the provided YOLO-based example and have also attempted to adapt the existing Hugging Face BERT example, though understandably, neither was directly applicable to my object detection model.

Could you provide some guidance or point me towards the recommended approach for integrating a custom DETR model for object detection within the Label Studio ML backend? Are there any specific patterns or examples you would suggest reviewing beyond the above mentioned ones?

Any insights 💡 or recommendations would be greatly appreciated!

Thanks a lot,
Chris

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.