Python- ocr识别模型(MNN模型)预测

2024-06-04 18:28
文章标签 python 模型 预测 识别 ocr mnn

本文主要是介绍Python- ocr识别模型(MNN模型)预测,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

import MNN
import cv2
import numpy as np
import timedef normlize_with_pad(img, width, height, ratio):h, w = img.shape[:2]  # 长边缩放为min_sideif h / w > height / width * ratio:val = int(img[-1,-1])try:img = cv2.resize(img, (int(ratio * height * w // h), height))img = cv2.copyMakeBorder(img, 0, 0, 0, (width - int(ratio * height * w // h)),cv2.BORDER_CONSTANT, value=[0, 0, 0])except Exception as e:print('error image shape {} {}'.format(h, w))img = cv2.resize(img, (width, height))return imgdef process(image_data, size):image_resize = normlize_with_pad(image_data, size[1],size[0], 1)input_data = np.array(image_resize)# input_data = np.ascontiguousarray(input_data)input_data = input_data.astype(np.float32)input_data = input_data / 255input_data = np.expand_dims(input_data, 0)input_data = np.expand_dims(input_data, 0)return input_datadef decode_out(str_index,logit, characters):char_list = []char_logit = []for i in range(len(str_index)):if str_index[i] != 0 and (not (i > 0 and str_index[i - 1] == str_index[i])):char_list.append(characters[str_index[i]-1])char_logit.append(logit[i].numpy())# char_l=1# for charl in char_logit:#     char_l*=charl# # print(char_l)# if not type(char_l)==int:#     char_l=char_l.numpy()#     if char_l.ndim>0:#         char_l=char_l[0]# print(char_logit)char_l=np.mean(char_logit)return ''.join(char_list),char_lif __name__ == "__main__":import torchmodel_path = 'densenet_rnn.mnn'# image_path = '61c785d7180e455aa6a7f892a44b733f_0_1713158555.jpg'image_path='OCRAExtended/0.jpg'resize = (32, 320)# characters= '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz[!"#$%&()*+.,/:;<=>?@\\^-_`{|}~]'characters= '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz/:-'# characters= '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ't1 = time.time()# (1) load modelnet = MNN.nn.load_module_from_file(model_path,  ["images"], ["outputs"])# net = MNN.nn.load_module_from_file(model_path,  ["Input:0"], ["model/swin_tiny_patch4_window7_224/out/truediv:0"])# preprocessprint(image_path)image_data = cv2.imread(image_path,0)input_data = process(image_data, resize)# (2) 构建一个Var类型的占位符来保存numpy,placeholder(shape, format, dtype)# print(input_data.shape) #(1, 1, 320, 32)input_var = MNN.expr.placeholder(input_data.shape, MNN.expr.NCHW)input_var.write(input_data)# (3) cv2 read shape is NHWC, Module's need is NC4HW4, convert itinput_var = MNN.expr.convert(input_var, MNN.expr.NC4HW4)# print(input_var.shape)# (4) inferenceoutput_var = net.forward(input_var)# print(output_var.shape) #[1, 160, 37]# (5) the output from net may be NC4HW4, turn to linear layout# output_var = MNN.expr.convert(output_var, MNN.expr.NCHW)# print(output_var.shape)output_var = output_var.read()output_var = torch.tensor(output_var)# print(output_var.shape) #(80, 1, 66) #torch.Size([1, 160, 77])logit, preds = output_var.max(2)logit = torch.exp(logit)preds = preds.transpose(1, 0).contiguous().view(-1)# print(preds)lab2str,char_logit = decode_out(preds,logit,characters )# lab2str,char_logit = decode_out(preds,logit[0],characters )print(lab2str,char_logit)t2 = time.time()# ./MNNConvert -f ONNX --modelFile "kang-slim.onnx" --MNNModel "kang-slim.mnn" --bizCode MNN

参考文章:Ubuntu18.04上MNN编译与使用(Python版)_mnn使用python cpu推理 demo-CSDN博客

这篇关于Python- ocr识别模型(MNN模型)预测的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1030771

相关文章

Python函数作用域示例详解

《Python函数作用域示例详解》本文介绍了Python中的LEGB作用域规则,详细解析了变量查找的四个层级,通过具体代码示例,展示了各层级的变量访问规则和特性,对python函数作用域相关知识感兴趣... 目录一、LEGB 规则二、作用域实例2.1 局部作用域(Local)2.2 闭包作用域(Enclos

Python实现对阿里云OSS对象存储的操作详解

《Python实现对阿里云OSS对象存储的操作详解》这篇文章主要为大家详细介绍了Python实现对阿里云OSS对象存储的操作相关知识,包括连接,上传,下载,列举等功能,感兴趣的小伙伴可以了解下... 目录一、直接使用代码二、详细使用1. 环境准备2. 初始化配置3. bucket配置创建4. 文件上传到os

使用Python实现可恢复式多线程下载器

《使用Python实现可恢复式多线程下载器》在数字时代,大文件下载已成为日常操作,本文将手把手教你用Python打造专业级下载器,实现断点续传,多线程加速,速度限制等功能,感兴趣的小伙伴可以了解下... 目录一、智能续传:从崩溃边缘抢救进度二、多线程加速:榨干网络带宽三、速度控制:做网络的好邻居四、终端交互

Python中注释使用方法举例详解

《Python中注释使用方法举例详解》在Python编程语言中注释是必不可少的一部分,它有助于提高代码的可读性和维护性,:本文主要介绍Python中注释使用方法的相关资料,需要的朋友可以参考下... 目录一、前言二、什么是注释?示例:三、单行注释语法:以 China编程# 开头,后面的内容为注释内容示例:示例:四

Python中win32包的安装及常见用途介绍

《Python中win32包的安装及常见用途介绍》在Windows环境下,PythonWin32模块通常随Python安装包一起安装,:本文主要介绍Python中win32包的安装及常见用途的相关... 目录前言主要组件安装方法常见用途1. 操作Windows注册表2. 操作Windows服务3. 窗口操作

Python中re模块结合正则表达式的实际应用案例

《Python中re模块结合正则表达式的实际应用案例》Python中的re模块是用于处理正则表达式的强大工具,正则表达式是一种用来匹配字符串的模式,它可以在文本中搜索和匹配特定的字符串模式,这篇文章主... 目录前言re模块常用函数一、查看文本中是否包含 A 或 B 字符串二、替换多个关键词为统一格式三、提

python常用的正则表达式及作用

《python常用的正则表达式及作用》正则表达式是处理字符串的强大工具,Python通过re模块提供正则表达式支持,本文给大家介绍python常用的正则表达式及作用详解,感兴趣的朋友跟随小编一起看看吧... 目录python常用正则表达式及作用基本匹配模式常用正则表达式示例常用量词边界匹配分组和捕获常用re

python实现对数据公钥加密与私钥解密

《python实现对数据公钥加密与私钥解密》这篇文章主要为大家详细介绍了如何使用python实现对数据公钥加密与私钥解密,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录公钥私钥的生成使用公钥加密使用私钥解密公钥私钥的生成这一部分,使用python生成公钥与私钥,然后保存在两个文

python删除xml中的w:ascii属性的步骤

《python删除xml中的w:ascii属性的步骤》使用xml.etree.ElementTree删除WordXML中w:ascii属性,需注册命名空间并定位rFonts元素,通过del操作删除属... 可以使用python的XML.etree.ElementTree模块通过以下步骤删除XML中的w:as

使用Python绘制3D堆叠条形图全解析

《使用Python绘制3D堆叠条形图全解析》在数据可视化的工具箱里,3D图表总能带来眼前一亮的效果,本文就来和大家聊聊如何使用Python实现绘制3D堆叠条形图,感兴趣的小伙伴可以了解下... 目录为什么选择 3D 堆叠条形图代码实现:从数据到 3D 世界的搭建核心代码逐行解析细节优化应用场景:3D 堆叠图