6 lines
185 B
Bash
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
|
||
|
|
|