基于CTPN（tensorflow）+CRNN（pytorch）+CTC的不定长文本检测和识别

最新推荐文章于 2023-04-02 11:51:32 发布

一只小白___

最新推荐文章于 2023-04-02 11:51:32 发布

阅读量1.1k

点赞数 2

本文链接：https://blog.youkuaiyun.com/weixin_41597991/article/details/103551189

版权

文章主要是为了说明在windows平台下调试遇到的问题。

代码地址：https://github.com/Aurora11111/chinese-ocr-pytorch

按照相关说明，安装相关的pytorch和tensorflow，我使用的是python3.7

下载代码后，需要进行修改才能在windows上运行。

1.修改cython_nms.pyx文件中的第25行，修改为如下

cdef np.ndarray[np.int_t, ndim=1] order = scores.argsort()[::-1].astype('int32')

2.新建一个python文件，假如为T.py，内容如下，主要为了解决pyx文件无法在windows下调用的问题。

import sys
import numpy as np

A = sys.path.insert(0, "..")
from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
from Cython.Distutils import build_ext

# ext_module = cythonize("TestOMP.pyx")
ext_module = Extension(
    "cython_nms",
    ["cython_nms.pyx"],
    extra_compile_args=["/openmp"],
    extra_link_args=["/openmp"],
)

setup(
    cm