wangjin0928
/
AIlib2
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 Commits
6 Branches
Tree: efcf713dcc
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'efcf713dcc'
${ noResults }
AIlib2/segutils/run.sh

9 lines
350B
Raw Blame History 

  1. gpu=2080Ti

  2. name=cityMangement3

  3. nclass=2

  4. mWidth=640

  5. mHeight=360

  6. python segmodel_trt.py --weights /mnt/thsw2/DSP2/weights/${name}/stdc_360X640.pth --nclass ${nclass} --mWidth ${mWidth} --mHeight ${mHeight}

  7. 

  8. mv /mnt/thsw2/DSP2/weights/${name}/stdc_${mWidth}X${mHeight}.engine /mnt/thsw2/DSP2/weights/${name}/stdc_${mHeight}X${mWidth}_${gpu}_fp16.engine