3090-2080Ti

demo.py

@@ -18,12 +18,14 @@ def river_demo_v2():
     ###注意TRT模型生成时，就需要对应cuda device，下面的trt文件是cuda:0生成的，device只能是0  
     ##以下参数目前不可改
     
-    Detweights = "../AIlib/weights/yolov5/class8/bestcao_fp16.engine" ##升级后的检测模型
+    gpuname='3090'
+    Detweights = "../AIlib/weights/yolov5/class8/bestcao_%s_fp16.engine"%(gpuname) ##升级后的检测模型
     labelnames = "../AIlib/weights/yolov5/class8/labelnames.json" ##对应类别表
+
     
     seg_nclass = 2 
     #Segweights = "../AIlib/weights/BiSeNet/checkpoint.pth"
-    Segweights = '../AIlib/weights/STDC/model_maxmIOU75_1720_0.946_360640.engine' ##升级的分割模型
+    Segweights = '../AIlib/weights/STDC/model_maxmIOU75_1720_0.946_360640_%s_fp16.engine'%(gpuname) ##升级的分割模型
     
     postFile= '../AIlib/conf/para.json'
     digitFont= { 'line_thickness':2,'boxLine_thickness':1, 'fontSize':1.0,'waterLineColor':(0,255,255),'waterLineWidth':3}