8 lines
351 B
Bash
8 lines
351 B
Bash
#gpu=2080Ti mWidth=448 mHeight=32 mode=en pth=/mnt/thsw2/DSP2/weights/ocr2/crnn_en
|
|
|
|
gpu=2080Ti mWidth=192 mHeight=32 mode=ch pth=/mnt/thsw2/DSP2/weights/ocr2/crnn_ch
|
|
python pth2onnx.py --weights ${pth}.pth --mode ${mode} --mWidth ${mWidth} --mHeight ${mHeight}
|
|
|
|
|
|
mv ${pth}_${mWidth}X${mHeight}.engine ${pth}_${gpu}_fp16_${mWidth}X${mHeight}.engine
|