|
|
|
|
|
|
|
|
'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 ,目前都没有在用
|
|
|
|
|
|
|