Update object_detectors.md

This commit is contained in:
Nicolas Mowen 2025-04-16 13:37:33 -06:00 committed by GitHub
parent c7573ca36e
commit 760af9ba67
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -990,6 +990,6 @@ bin/pip install onnx onnxruntime onnx-simplifier>=0.4.1
wget -O yolov9-t.pt "https://github.com/WongKinYiu/yolov9/releases/download/v0.1/yolov9-t-converted.pt" # download the weights
# prepare and run export script
sed -i "s/ckpt = torch.load(attempt_download(w), map_location='cpu')/ckpt = torch.load(attempt_download(w), map_location='cpu', weights_only=False)/g" <your_file>
sed -i "s/ckpt = torch.load(attempt_download(w), map_location='cpu')/ckpt = torch.load(attempt_download(w), map_location='cpu', weights_only=False)/g" ./models/experimental.py
python3 export.py --weights ./yolov9-t.pt --imgsz 320 --simplify --include onnx
```