AIlib2/segutils/run.sh

9 lines
350 B
Bash

gpu=2080Ti
name=cityMangement3
nclass=2
mWidth=640
mHeight=360
python segmodel_trt.py --weights /mnt/thsw2/DSP2/${name}/weights/stdc_360X640.pth --nclass ${nclass} --mWidth ${mWidth} --mHeight ${mHeight}
mv /mnt/thsw2/DSP2/weights/${name}/stdc_${mWidth}X${mHeight}.engine /mnt/thsw2/DSP2/weights/${name}/stdc_${mHeight}X${mWidth}_${gpu}_fp16.engine