roboflow / inference

A fast, easy-to-use, production-ready inference server for computer vision supporting deployment of many popular model architectures and fine-tuned models.

https://inference.roboflow.com

Other

1.3k stars 116 forks source link

Added RTDETR model to inference #558

Open Bhavay-2001 opened 2 months ago

Bhavay-2001 commented 2 months ago

Description

PR for #546

Type of change

Please delete options that are not relevant.

[ ] Bug fix (non-breaking change which fixes an issue)
[x] New feature (non-breaking change which adds functionality)
[x] This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

Any specific deployment considerations

Docs

[ ] Docs updated? What were the changes:

grzegorz-roboflow commented 2 months ago

Hi @Bhavay-2001, thank you for providing this PR! I'd love to test it, can you share python code with inference performed using this model?

Bhavay-2001 commented 2 months ago

Hi @grzegorz-roboflow, I haven't tested it myself. I got the code from here. I have created the notebook that test this code.

I have just tried to convert that code to a file.

Bhavay-2001 commented 2 months ago

Hi @grzegorz-roboflow, pls let me know what changes needs to be done.

Bhavay-2001 commented 2 months ago

Hi @probicheaux, can you pls provide a detailed review? Thanks

Bhavay-2001 commented 1 month ago

Hi @grzegorz-roboflow, PR is ready for review. Can you please check this. Thanks

yeldarby commented 1 month ago

Hey @Bhavay-2001! We greatly appreciate your contribution and thank you so much for submitting this PR.

I talked with the team and we are planning on incorporating fine-tuning of RT-DETR more tightly into Roboflow shortly so we need to adapt the inference integration to be compatible with the output of our training process.

This PR is a great start & we’ll likely continue working from it towards a release but we won’t be able to do that for a little while until the backend is more concrete. Best path forward in the meantime would be to use it via a plugin or fork. (Apologies for the inconvenience; we’ll update you when we have more to share on our end!)