21 lines
872 B
Python
21 lines
872 B
Python
import cv2
|
|
|
|
from db import MysqlSession
|
|
from db.mysql import ZxIeOcrerror
|
|
from photo_mask_error_check import check_error
|
|
from ucloud import ufile
|
|
from util import image_util
|
|
|
|
if __name__ == '__main__':
|
|
session = MysqlSession()
|
|
ocr_errors = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress)
|
|
.filter(ZxIeOcrerror.checktime.is_(None)).order_by(ZxIeOcrerror.pk_phrec.desc()).limit(50).all())
|
|
session.close()
|
|
|
|
for ocr_error in ocr_errors:
|
|
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)
|
|
image = check_error(ocr_error)
|
|
cv2.imwrite(f"./mask_optimization_result/{ocr_error.cfjaddress}.jpg", image)
|