test

demo3.0.py

@@ -257,7 +257,7 @@ def det_track_demo(business ):
             'postFile': '../AIlib2/weights/conf/%s/para.json'%('highWay2' ),###后处理参数文件
             'txtFontSize':20,###文本字符的大小
             'digitFont': { 'line_thickness':2,'boxLine_thickness':1, 'fontSize':1.0,'waterLineColor':(0,255,255),'segLineShow':True,'waterLineWidth':2},###显示框、线设置
-            'testImgPath':'/home/chenyukun/video',###测试图像的位置
+            'testImgPath':'/mnt/thsw2/DSP2/videos/frames/',###测试图像的位置
             'testOutPath':'images/results/',###输出测试图像位置
         }
         par['segPar']['mixFunction']['pars']['modelSize'] = par['segPar']['modelSize']
@@ -564,8 +564,8 @@ def det_track_demo(business ):
         time00 = time.time()
         retResults,timeOut = AI_det_track_batch(im0s, [i] ,modelPar,processPar,sort_tracker ,trackPar,segPar)
         #retResults,timeInfos = AI_det_track_batch(imgarray_list, iframe_list ,par0['modelPar'],par0['processPar'],par0['sort_tracker'] ,par0['trackPar'],segPar=par0['segPar'])
-        
-        drawBoxTraceSimplied(retResults[1],i, retResults[0][0],rainbows=rainbows,boxFlag=True,traceFlag=False)
+        if len(retResults[1])>0:
+            drawBoxTraceSimplied(retResults[1],i, retResults[0][0],rainbows=rainbows,boxFlag=True,traceFlag=False)
         time11 = time.time()
         image_array = retResults[0][0]
         '''