AIlib2/ocrUtils2/run.sh

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