search

naseemap47 / YOLO-NAS

Train and Inference your custom YOLO-NAS model by Single Command Line
Apache License 2.0
98 stars 13 forks source link

Yolo-nas change image size and pt save #23

Closed john09282922 closed 1 year ago

john09282922 commented 1 year ago

Hi, I would like to change image size of model when training Yolonas. And also save pt file, not ckpt file. Is it possible?

naseemap47 commented 1 year ago

Hi @john09282922 , To change image size, try with official colab build by super-gradients https://colab.research.google.com/drive/1yHrHkUR1X2u2FjjvNMfUbSXTkUul6o1P?usp=sharing#scrollTo=fBZenZegj9hm\

The Trainer class build by super_gradients, saving the model weight in the .ckpt format not .pt. But its fine to use .ckpt format. It have same purpose as .pt. Thank you.

naseemap47 commented 1 year ago

Hi @john09282922 , I added option to change input image size. Please look into it.

model .pt have meta data (class names, model type, size, etc..) but .pth don't have. But super-gradients didn't give feature for that. I also asked them for this. Currently not available. If you find any please let me know. But to over come this problem (class names), I manually added dummy loading for this.

If you have any questions and any improvements needed please let me know. Thank you.