Browse Source

fix bug

master
wangjin0928 1 year ago
parent
commit
1196606562
3 changed files with 3 additions and 2 deletions
  1. +3
    -2
      demo.py
  2. BIN
      images/drowning/luoshui1.png
  3. BIN
      images/test/垃圾_20220713_牛王庙沟_9240_9271.jpg

+ 3
- 2
demo.py View File

'device':'0', ###显卡号,如果用TRT模型,只支持0(单显卡) 'device':'0', ###显卡号,如果用TRT模型,只支持0(单显卡)
'labelnames':"../AIlib2/weights/conf/river/labelnames.json", ###检测类别对照表 'labelnames':"../AIlib2/weights/conf/river/labelnames.json", ###检测类别对照表
'max_workers':1, ###并行线程数 'max_workers':1, ###并行线程数
'Detweights':"../weights/%s/AIlib2/%s/yolov5_%s_fp16.engine"%(opt['gpu'], opt['business'] ,opt['gpu'] ),###检测模型路径
#'Detweights':"../weights/%s/AIlib2/%s/yolov5_%s_fp16.engine"%(opt['gpu'], opt['business'] ,opt['gpu'] ),###检测模型路径
'Detweights':"/mnt/thsw2/DSP2/weights/river/yolov5.pt",###检测模型路径
'detModelpara':[{"id":str(x),"config":{"k1":"v1","k2":"v2"}} for x in [0,1,2,3,4,5,6,7,8,9] ],###控制哪些检测类别显示,输出 'detModelpara':[{"id":str(x),"config":{"k1":"v1","k2":"v2"}} for x in [0,1,2,3,4,5,6,7,8,9] ],###控制哪些检测类别显示,输出
'seg_nclass':2,###分割模型类别数目,默认2类 'seg_nclass':2,###分割模型类别数目,默认2类
'segRegionCnt':1,###分割模型结果需要保留的等值线数目 'segRegionCnt':1,###分割模型结果需要保留的等值线数目
#jkm_demo() #jkm_demo()
businessAll=['river2','AnglerSwimmer', 'countryRoad','forest2', 'pedestrian' , 'smogfire' , 'vehicle','ship2',"highWay2","channelEmergency","cityMangement","drowning","noParking"] businessAll=['river2','AnglerSwimmer', 'countryRoad','forest2', 'pedestrian' , 'smogfire' , 'vehicle','ship2',"highWay2","channelEmergency","cityMangement","drowning","noParking"]
#trafficAccident、出问题 #trafficAccident、出问题
businessAll = ["river"]
businessAll = ['drowning']
# forest 、 ocr2 、ocr_en 、 river 、 road 、 ship ,目前都没有在用 # forest 、 ocr2 、ocr_en 、 river 、 road 、 ship ,目前都没有在用

BIN
images/drowning/luoshui1.png View File

Before After
Width: 800  |  Height: 431  |  Size: 563KB

BIN
images/test/垃圾_20220713_牛王庙沟_9240_9271.jpg View File

Before After
Width: 1920  |  Height: 1080  |  Size: 1.0MB

Loading…
Cancel
Save