添加耗时日志
This commit is contained in:
@@ -82,12 +82,16 @@ 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)
|
||||
|
||||
Reference in New Issue
Block a user