fix bug

demo.py

@@ -135,7 +135,8 @@ def detSeg_demo(opt):
             'device':'0', ###显卡号,如果用TRT模型，只支持0（单显卡）
             'labelnames':"../AIlib2/weights/conf/river/labelnames.json", ###检测类别对照表
             '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]  ],###控制哪些检测类别显示，输出
             'seg_nclass':2,###分割模型类别数目，默认2类
             'segRegionCnt':1,###分割模型结果需要保留的等值线数目       
@@ -941,7 +942,7 @@ if __name__=="__main__":
     #jkm_demo()
     businessAll=['river2','AnglerSwimmer',  'countryRoad','forest2',  'pedestrian' , 'smogfire' , 'vehicle','ship2',"highWay2","channelEmergency","cityMangement","drowning","noParking"]
     #trafficAccident、出问题
-    businessAll = ["river"]
+    businessAll = ['drowning']
     
     # forest 、 ocr2  、ocr_en 、 river 、 road 、 ship ,目前都没有在用