增加检测矫正后用时上升比较明显,常规启用photo_review_6来弥补

This commit is contained in:
2024-08-28 14:31:46 +08:00
parent 0e67e01723
commit 3f6384a2fb
3 changed files with 19 additions and 33 deletions

View File

@@ -82,16 +82,12 @@ def information_extraction(ie, phrecs, identity):
image = image_util.read(img_path)
target_images = []
det_time = time.time()
target_images += detector.get_book_areas(image) # 识别文档区域并裁剪
logging.info(f"det耗时{time.time() - det_time}")
if not target_images:
target_images.append(image) # 识别失败
angle_count = defaultdict(int, {"0": 0}) # 分割后图片的最优角度统计
for target_image in target_images:
dewarp_time = time.time()
dewarped_image = dewarp.dewarp_image(target_image) # 去扭曲
logging.info(f"dewarp耗时{time.time() - dewarp_time}")
angles = image_util.parse_rotation_angles(dewarped_image)
zx_ie_results = []
split_results = image_util.split(dewarped_image)