diff --git a/demo.py b/demo.py index bde0035..ff935cb 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,'waterLineColor':(0,255,255),'waterLineWidth':3} + digitFont= { 'line_thickness':2,'boxLine_thickness':1, 'fontSize':1.0,'waterLineColor':(0,255,255),'waterLineWidth':3} conf_thres,iou_thres,classes,rainbows=get_postProcess_para(postFile) ####模型选择参数用如下: