AIlib2/obbUtils/pth2onnx.sh

6 lines
185 B
Bash

gpu=2080Ti
weights=/mnt/thsw2/DSP2/weights/ship2/obb_608X608
#python pth2onnx.py --weights ${weights}.pth --mWidth 608 --mHeight 608
mv ${weights}.engine ${weights}_${gpu}_fp16.engine