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-08-28 14:04:11 +08:00
parent 8085c75207
commit 0e67e01723
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yml
View File

@@ -1,6 +1,6 @@
x-env: x-env:
&template &template
image: fcb_photo_review:1.13.5 image: fcb_photo_review:1.13.6
restart: always restart: always
services: services:

4
photo_review/auto_photo_review.py
View File

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