From da5d22f76329e062abf8624446d8795f4b627a0c Mon Sep 17 00:00:00 2001
From: thsw <wangjin0928@163.com>
Date: Sat, 8 Oct 2022 16:09:44 +0800
Subject: [PATCH] digitSize

---
 demo.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/demo.py b/demo.py
index 9b8f876..bf66fb0 100644
--- a/demo.py
+++ b/demo.py
@@ -26,7 +26,7 @@ def river_demo():
     Segweights = '../AIlib/weights/STDC/model_maxmIOU75_1720_0.946_360640.pth' ##升级的分割模型
     
     postFile= '../AIlib/conf/para.json'
-    
+    digitFont= { 'line_thickness':2, 'fontSize':1.0} 
     conf_thres,iou_thres,classes,rainbows=get_postProcess_para(postFile)
    
     ####模型选择参数用如下：
@@ -67,7 +67,7 @@ def river_demo():
         imgpath = os.path.join(impth, folders[i])
         im0s=[cv2.imread(imgpath)]
         time00 = time.time()
-        p_result,timeOut = AI_process(im0s,model,segmodel,names,label_arraylist,rainbows,half,device,conf_thres, iou_thres,allowedList)
+        p_result,timeOut = AI_process(im0s,model,segmodel,names,label_arraylist,rainbows,half,device,conf_thres, iou_thres,allowedList,font=digitFont)
         time11 = time.time()
         image_array = p_result[1]
         cv2.imwrite(  os.path.join( outpth,folders[i] ) ,image_array )
@@ -153,6 +153,6 @@ def road_forest_demo(business ):
     
         
 if __name__=="__main__":
-    #river_demo()
-    road_forest_demo('forest'  )
+    river_demo()
+    #road_forest_demo('forest'  )