tannedcung/sbt-idp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update process_img parameter

Browse Source
This commit is contained in:
PhanThanhTrung 2024-03-12 14:49:17 +07:00
parent c3d177dc0c
commit b2ee0cad41
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cope2n-ai-fi/api/sdsap_sbt/prediction_sbt.py
View File

@ -38,7 +38,7 @@ def sbt_predict(image_url, engine) -> None:
os.makedirs(save_dir, exist_ok = True) os.makedirs(save_dir, exist_ok = True)
tmp_image_path = os.path.join(save_dir, f"{uuid.uuid4()}.jpg") tmp_image_path = os.path.join(save_dir, f"{uuid.uuid4()}.jpg")
cv2.imwrite(tmp_image_path, img) cv2.imwrite(tmp_image_path, img)
outputs = process_img(img_path=tmp_image_path, outputs = process_img(img=tmp_image_path,
save_dir=save_dir, save_dir=save_dir,
engine=engine, engine=engine,
export_all=False, # False export_all=False, # False
@ -71,7 +71,6 @@ def predict(page_numb, image_url):
""" """
sbt_result = sbt_predict(image_url, engine=sbt_engine) sbt_result = sbt_predict(image_url, engine=sbt_engine)
print(sbt_result)
output_dict = { output_dict = {
"document_type": "invoice", "document_type": "invoice",
"document_class": " ", "document_class": " ",