9 lines
350 B
Bash
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
|