diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py index b5f7390..ea10e73 100755 --- a/ieemoo-ai-isempty.py +++ b/ieemoo-ai-isempty.py @@ -134,32 +134,32 @@ def get_isempty(): imgdata = base64.b64decode(pic) imgdata_np = np.frombuffer(imgdata, dtype='uint8') img_src = cv2.imdecode(imgdata_np, cv2.IMREAD_COLOR) - #cv2.imwrite('huanyuan.jpg',img_src) - img_data = Image.fromarray(np.uint8(img_src)) #这个转换不能要,会导致判空错误增加 - #img_data = Image.open('huanyuan.jpg') + cv2.imwrite('huanyuan.jpg',img_src) + #img_data = Image.fromarray(np.uint8(img_src)) #这个转换不能要,会导致判空错误增加 + img_data = Image.open('huanyuan.jpg') result = predictor.normal_predict(img_data, result) # 1==empty, 0==nonEmpty - #riseresult = lightrise.riseempty(img_data) + riseresult = lightrise.riseempty(img_data) #print(riseresult["rst_cls"]) - # if(result["rst_cls"]==1): - # if(riseresult["rst_cls"]==1): - # result = {} - # result["success"] = "true" - # result["rst_cls"] = 1 - # else: - # result = {} - # result["success"] = "true" - # result["rst_cls"] = 0 - # else: - # if(riseresult["rst_cls"]==0): - # result = {} - # result["success"] = "true" - # result["rst_cls"] = 0 - # else: - # result = {} - # result["success"] = "true" - # result["rst_cls"] = 1 + if(result["rst_cls"]==1): + if(riseresult["rst_cls"]==1): + result = {} + result["success"] = "true" + result["rst_cls"] = 1 + else: + result = {} + result["success"] = "true" + result["rst_cls"] = 0 + else: + if(riseresult["rst_cls"]==0): + result = {} + result["success"] = "true" + result["rst_cls"] = 0 + else: + result = {} + result["success"] = "true" + result["rst_cls"] = 1 return repr(result)