From bbdeaa94d875286e12505cf900b6d13f867b8e42 Mon Sep 17 00:00:00 2001 From: thsw Date: Sun, 14 Aug 2022 20:56:03 +0800 Subject: [PATCH] update --- demo.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/demo.py b/demo.py index b503536..642f14b 100644 --- a/demo.py +++ b/demo.py @@ -1,7 +1,7 @@ import sys sys.path.extend(['..','../AIlib' ]) -from AI import AI_process,get_postProcess_para +from AI import AI_process,AI_process_forest,get_postProcess_para import cv2,os,time from segutils.segmodel import SegModel from models.experimental import attempt_load @@ -130,7 +130,8 @@ def forest_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_forest(im0s,model,segmodel,names,label_arraylist,rainbows,half,device,conf_thres, iou_thres,allowedList) time11 = time.time() image_array = p_result[1] cv2.imwrite( os.path.join( outpth,folders[i] ) ,image_array )