This commit is contained in:
Nicolas Mowen 2025-05-06 07:52:59 -06:00
parent 50e81f0a7f
commit 3666a80677

View File

@ -1069,5 +1069,5 @@ wget -O yolov9-t.pt "https://github.com/WongKinYiu/yolov9/releases/download/v0.1
# prepare and run export script # 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" ./models/experimental.py 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 bin/python3 export.py --weights ./yolov9-t.pt --imgsz 320 --simplify --include onnx
``` ```