Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

统一模型接口，新增文本分类接口

Browse Source
This commit is contained in:
liuyebo
2024-09-27 13:50:55 +08:00
parent 117b29a737
commit f1149854ce
13 changed files with 144 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
photo_review/auto_photo_review.py
View File

@@ -185,8 +185,8 @@ def information_extraction(ie, phrecs, identity):
target_images = model_util.request_book_areas(img_path) # 识别文档区域并裁剪
angle_count = defaultdict(int, {'0': 0}) # 分割后图片的最优角度统计
for target_image in target_images:
dewarped_image = model_util.request_dewarped_image(target_image) # 去扭曲
angles = model_util.request_image_orientation(dewarped_image)
dewarped_image = model_util.dewarp(target_image) # 去扭曲
angles = model_util.clas_orientation(dewarped_image)
split_results = image_util.split(dewarped_image)
for split_result in split_results: