10 lines
456 B
Bash
10 lines
456 B
Bash
#动态尺寸的stdc转trt没有成功，因为里有用到了全局池化，池化的过程中kener_size不固定
|
||
gpu=2080Ti
|
||
name=crackMeasurement
|
||
nclass=2
|
||
mWidth=0
|
||
mHeight=0
|
||
python toTrt.py --weights /mnt/thsw2/DSP2/weights/${name}/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
|