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