search

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.39k stars 344 forks source link

Yolonas invalid weights file #407

Closed storm12t48 closed 12 months ago

storm12t48 commented 12 months ago

Good morning I can't generate a weight and label for a custom model I always have the same invalid weight error what weight should I take knowing that I don't have the same class number or name as a classic model? but i can generate classic model 

python3 export_yolonas.py -m ckpt_best -w ckpt_best.pth  --simplify
The console stream is logged into /root/sg_logs/console.log
[2023-07-10 08:05:22] INFO - crash_tips_setup.py - Crash tips is enabled. You can set your environment variable to CRASH_HANDLER=FALSE to disable it
[2023-07-10 08:05:31] WARNING - __init__.py - Failed to import pytorch_quantization
[2023-07-10 08:05:31] WARNING - calibrator.py - Failed to import pytorch_quantization
[2023-07-10 08:05:31] WARNING - export.py - Failed to import pytorch_quantization
[2023-07-10 08:05:31] WARNING - selective_quantization_utils.py - Failed to import pytorch_quantization
Invalid weights file
storm12t48 commented 12 months ago

its ok right command its : python3 export_yolonas.py -m yolo_nas_s -w ckpt_best.pth --simplify