18 lines
637 B
Python
18 lines
637 B
Python
import cv2
|
|
|
|
from db import MysqlSession
|
|
from db.mysql import ZxIeOcrerror
|
|
from photo_mask_error_check import check_error
|
|
|
|
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:
|
|
image = check_error(ocr_error)
|
|
if image is None:
|
|
continue
|
|
cv2.imwrite(f"./mask_optimization_result/{ocr_error.cfjaddress}.jpg", image)
|