super-gradients icon indicating copy to clipboard operation
super-gradients copied to clipboard

Integrating custom models into backbone

Open mustafaAlgun opened this issue 2 years ago • 2 comments

Hello there!

I loved your work, keep it up!

I'd like to integrate a custom attention model as backbone model for an object detection task and test it out. Is there any documentation or tutorial that you can provide so that I can follow? Or any help would be appreciated!

mustafaAlgun avatar May 04 '23 09:05 mustafaAlgun

Hey @mustafaAlgun,

We have yet to add a tutorial on customizing the YoloNAS backbone. We do plan on making customiaztion easier though. In the meanwhile, you can check our CustomizableDetector class docs which is the parent class for YoloNAS to see how you can adapt your backbone code so it would fit there. I am here to help if any question arises (:

shaydeci avatar May 04 '23 10:05 shaydeci