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-07-23 16:59:36 +08:00
parent 221075bd67
commit d2ea9ec521
5 changed files with 81 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
test_photo_mask_optimization.py
View File

@@ -2,7 +2,7 @@ import cv2
from db import MysqlSession
from db.mysql import ZxIeOcrerror
from photo_mask.photo_mask import handle_image_for_mask, get_mask_layout
from photo_mask_error_check import check_error
from ucloud import ufile
from util import image_util
@@ -16,30 +16,5 @@ if __name__ == '__main__':
final_img_url = ufile.get_private_url(ocr_error.cfjaddress, "drg100")
final_image = image_util.read(final_img_url)
cv2.imwrite(f"./mask_optimization_result/answer/{ocr_error.cfjaddress}.jpg", final_image)
img_url = ufile.get_private_url(ocr_error.cfjaddress, "drg2015")
if not img_url:
# 没有自动涂抹的图片
img_url = ufile.get_private_url(ocr_error.cfjaddress, "drg103")
name = ocr_error.cXm
id_card_num = ocr_error.cSfzh
image = image_util.read(img_url)
# image = image_util.rotate(image, 90)
split_results = image_util.split(image)
for split_result in split_results:
to_mask_img, x_offset, y_offset = handle_image_for_mask(split_result)
results = get_mask_layout(to_mask_img, name, id_card_num)
if results:
is_masked = True
for result in results:
result = (
result[0] + x_offset,
result[1] + y_offset,
result[2] + x_offset,
result[3] + y_offset,
)
cv2.rectangle(image, (int(result[0]), int(result[1])), (int(result[2]), int(result[3])),
(0, 0, 0), -1, 0)
image = check_error(ocr_error)
cv2.imwrite(f"./mask_optimization_result/{ocr_error.cfjaddress}.jpg", image)