Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.46k stars 481 forks source link

How to KD for object detection #1507

Closed minhhotboy9x closed 5 months ago

minhhotboy9x commented 9 months ago

💡 Your Question

I saw in this repo that there is knowledge distillation for different models. However, I got nothing when I tried to find a tutorial for object detection distillation. There is only a tutorial for classification distillation. Can somebody help me to KD for object detection in this repo, please?

Versions

No response

harpreetsahota204 commented 8 months ago

Hi @minhhotboy9x, it will follow the same pattern as the image classification tutorial. Have you tried it out? If so, what difficulty did you face?

BloodAxe commented 5 months ago

KD of object detection is currently not supported in SG.