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-24 08:36:34 +08:00
parent 9c21152823
commit 90a6d5ec75
6 changed files with 105 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
photo_review/auto_photo_review.py
View File

@@ -182,15 +182,11 @@ def information_extraction(ie, phrecs, identity):
result = merge_result(result, ie_result['result'])
else:
target_images = []
target_images += model_util.request_book_areas(img_path) # 识别文档区域并裁剪
if not target_images:
target_images.append(image) # 识别失败
target_images = model_util.request_book_areas(img_path) # 识别文档区域并裁剪
angle_count = defaultdict(int, {'0': 0}) # 分割后图片的最优角度统计
for target_image in target_images:
# dewarped_image = dewarp.dewarp_image(target_image) # 去扭曲
dewarped_image = target_image
angles = image_util.parse_rotation_angles(dewarped_image)
dewarped_image = model_util.request_dewarped_image(target_image) # 去扭曲
angles = model_util.request_image_orientation(dewarped_image)
split_results = image_util.split(dewarped_image)
for split_result in split_results: