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开发PDF转PNG的可视化工具

《基于Python开发PDF转PNG的可视化工具》在数字文档处理领域,PDF到图像格式的转换是常见需求,本文介绍如何利用Python的PyMuPDF库和Tkinter框架开发一个带图形界面的PDF转P... 目录一、引言二、功能特性三、技术架构1. 技术栈组成2. 系统架构javascript设计3.效果图

Python如何在Word中生成多种不同类型的图表

《Python如何在Word中生成多种不同类型的图表》Word文档中插入图表不仅能直观呈现数据,还能提升文档的可读性和专业性,本文将介绍如何使用Python在Word文档中创建和自定义各种图表,需要的... 目录在Word中创建柱形图在Word中创建条形图在Word中创建折线图在Word中创建饼图在Word

Python Excel实现自动添加编号

《PythonExcel实现自动添加编号》这篇文章主要为大家详细介绍了如何使用Python在Excel中实现自动添加编号效果,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录1、背景介绍2、库的安装3、核心代码4、完整代码1、背景介绍简单的说,就是在Excel中有一列h=会有重复

Python FastAPI入门安装使用

《PythonFastAPI入门安装使用》FastAPI是一个现代、快速的PythonWeb框架,用于构建API,它基于Python3.6+的类型提示特性,使得代码更加简洁且易于绶护,这篇文章主要介... 目录第一节:FastAPI入门一、FastAPI框架介绍什么是ASGI服务(WSGI)二、FastAP

Python中Windows和macOS文件路径格式不一致的解决方法

《Python中Windows和macOS文件路径格式不一致的解决方法》在Python中,Windows和macOS的文件路径字符串格式不一致主要体现在路径分隔符上,这种差异可能导致跨平台代码在处理文... 目录方法 1:使用 os.path 模块方法 2:使用 pathlib 模块(推荐)方法 3:统一使

一文教你解决Python不支持中文路径的问题

《一文教你解决Python不支持中文路径的问题》Python是一种广泛使用的高级编程语言,然而在处理包含中文字符的文件路径时,Python有时会表现出一些不友好的行为,下面小编就来为大家介绍一下具体的... 目录问题背景解决方案1. 设置正确的文件编码2. 使用pathlib模块3. 转换路径为Unicod

Python结合Flask框架构建一个简易的远程控制系统

《Python结合Flask框架构建一个简易的远程控制系统》这篇文章主要为大家详细介绍了如何使用Python与Flask框架构建一个简易的远程控制系统,能够远程执行操作命令(如关机、重启、锁屏等),还... 目录1.概述2.功能使用系统命令执行实时屏幕监控3. BUG修复过程1. Authorization

Python使用DeepSeek进行联网搜索功能详解

《Python使用DeepSeek进行联网搜索功能详解》Python作为一种非常流行的编程语言,结合DeepSeek这一高性能的深度学习工具包,可以方便地处理各种深度学习任务,本文将介绍一下如何使用P... 目录一、环境准备与依赖安装二、DeepSeek简介三、联网搜索与数据集准备四、实践示例:图像分类1.

Python中__new__()方法适应及注意事项详解

《Python中__new__()方法适应及注意事项详解》:本文主要介绍Python中__new__()方法适应及注意事项的相关资料,new()方法是Python中的一个特殊构造方法,用于在创建对... 目录前言基本用法返回值单例模式自定义对象创建注意事项总结前言new() 方法在 python 中是一个

Python批量调整Word文档中的字体、段落间距及格式

《Python批量调整Word文档中的字体、段落间距及格式》这篇文章主要为大家详细介绍了如何使用Python的docx库来批量处理Word文档,包括设置首行缩进、字体、字号、行间距、段落对齐方式等,需... 目录关键代码一级标题设置  正文设置完整代码运行结果最近关于批处理格式的问题我查了很多资料,但是都没