search

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.59k stars 510 forks source link

How to get pretrained weights of yolo-nas on obj365 datasets #1143

Closed Yi-dashi closed 1 year ago

Yi-dashi commented 1 year ago

💡 Your Question

How to get pretrained weights of yolo-nas_s on obj365 datasets? yolo-nas got a mAP of 47.5, is this the result of pre-training on the obj365 dataset and then fine-tuning on the coco dataset?

Versions

No response

BloodAxe commented 1 year ago

Objects365 was part of pretraining of Yolo-NAS. However we do not provide pretrained weights for Objects365, since there wasn't any goal to reach best possible accuracy on that dataset. So Yolo-NAS weights for Objects365 will not be released.

PavanNettam commented 1 year ago

Are pre-trained weights of yolo-nas_s on coco (or any other dataset) released?