add 4080

readme.md

@@ -1,2 +1,9 @@
 2.0--变化如下：
 每个模型都要采用TRT模型
+2023.1.17
+1.增加检测模型(countryRoad)--乡间道路检测---类别包括：违法种植
+2.增加钓鱼游泳模型(AnglerSwimmer)--类别包括：钓鱼、游泳
+3.更新车辆检测模型--增加网络公开卡口数据
+4.更新高速路里的公路检测分割模型--增加乡间小路的数据
+2023.1.29
+1.增加模型对4080的TRT兼容

segutils/run.sh

@@ -0,0 +1,7 @@
+for business in road river
+do
+	echo do $business
+	python segmodel_trt.py --weights ../../../yolov5TRT/weights/${business}/stdc_360X640.pth
+	rm ../../../yolov5TRT/weights/${business}/stdc_360X640.onnx
+	mv ../../../yolov5TRT/weights/${business}/stdc_360X640.engine ../../../yolov5TRT/weights/${business}/stdc_360X640_4080_fp16.engine
+done

segutils/segmodel_trt.py

@@ -1,4 +1,5 @@
 import torch
+import argparse
 import sys,os
 sys.path.extend(['segutils'])
 from core.models.bisenet import BiSeNet
@@ -521,10 +522,12 @@ def EngineInfer(par):
      
                                     
 if __name__=='__main__':
-    pthFile = Path('../weights/road/stdc_360X640.pth')
-   
-    
-    
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--weights', type=str, default='stdc_360X640.pth', help='model path(s)')
+    opt = parser.parse_args()
+    print( opt.weights  )
+    #pthFile = Path('../../../yolov5TRT/weights/river/stdc_360X640.pth')
+    pthFile = Path(opt.weights)
     onnxFile = pthFile.with_suffix('.onnx')
     trtFile =  onnxFile.with_suffix('.engine')