lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update ieemoo-ai-isempty.py.

Browse Source
This commit is contained in:
Brainway
2022-10-11 06:36:36 +00:00
committed by Gitee
parent 6703c1bcaa
commit bd8f639537
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
ieemoo-ai-isempty.py
View File

@ -124,16 +124,13 @@ def get_isempty():
result = {"success": "false", result = {"success": "false",
"rst_cls": '-1', "rst_cls": '-1',
} }
try:
imgdata = base64.b64decode(pic) imgdata = base64.b64decode(pic)
imgdata_np = np.frombuffer(imgdata, dtype='uint8') imgdata_np = np.frombuffer(imgdata, dtype='uint8')
img_src = cv2.imdecode(imgdata_np, cv2.IMREAD_COLOR) img_src = cv2.imdecode(imgdata_np, cv2.IMREAD_COLOR)
img_data = Image.fromarray(np.uint8(img_src)) img_data = Image.fromarray(np.uint8(img_src))
result = predictor.normal_predict(img_data, result) # 1==empty, 0==nonEmpty result = predictor.normal_predict(img_data, result) # 1==empty, 0==nonEmpty
except Exception as e:
print(e)
return repr(result)
print(repr(result))
return repr(result) return repr(result)
if __name__ == "__main__": if __name__ == "__main__":