由于要求提交模型预测压缩包,且数据集数据量太大,所以编写一个展示OCR识别效果的函数,方便预览。主体函数如下:
def convert_OCR_some(text):
for false_punc, true_punc in OCRwikisome.items():
correct = text.replace(false_punc, true_punc)
if correct != text:
return correct
return text
# 获得预测结果
def get_result(image: str) -> str:
result = ocr.ocr(image)
text = ""
for i in result[0]:
text += i[1][0] + '\n'
return text
# 设置图标路径(绝对路径)
img = "./imgs/8.jpg"
print(get_result(img))
预测结果如下:
1、清晰印刷体
2、清晰手写楷体
3、手写其他字体
4、模糊英文字体
5、模糊中文字体
6、模糊手写字体
7、颜色图片
等等,不再一一列举。