CTPN在windows的CPU上运行经验

最新推荐文章于 2021-01-07 18:12:24 发布

田野星

最新推荐文章于 2021-01-07 18:12:24 发布

阅读量4.1k

点赞数 2

分类专栏： tensorflow 文章标签： ctpn

本文链接：https://blog.csdn.net/qq_32791307/article/details/81324343

版权

本文详细介绍了如何在Windows上使用CPU运行CTPN，包括修改配置文件，处理编译错误，以及安装和运行的步骤。关键操作包括在text.yml和config.py中禁用GPU，调整cython_nms.pyx的类型，以及编译和复制必要的库文件。确保正确放置checkpoint文件以避免导入错误。参考了issues/73和43，适用于win8，TensorFlow 1.8，Python 3.6和easydict 1.7的环境。

摘要由CSDN通过智能技术生成

假设根目录在 C:\Users\uset\text-detection-ctpn-master

设置 "USE_GPU_NMS " 在文件 ./ctpn/text.yml as "False"
设置 "__C.USE_GPU_NMS" 在文件 ./lib/fast_rcnn/config.py 为 "False";
注释掉 "from lib.utils.gpu_nms import gpu_nms" 在文件 ./lib/fast_rcnn/nms_wrapper.py 中;
在文件lib \ utils \ cython_nms.pyx的第25行进行一些更改“np.int_t”更改为“np.intp_t”
否则出现“ValueError：Buffer dtype mismatch，expected'int_t'但得到'long long' “在第6步。
注释掉 "from . import gpu_nms" 在文件 .\lib\utils_init.py 中;
直接运行 make.sh 在文件 .\lib\utils 中会出现 bbox.cp36-win_amd64.pyd和cython_nms.cp36-win_amd64.pyd
或者重构setup 并运行python setup.py build 然后把build 文件夹中的 bbox.cp36-win_amd64.pyd和cython_nms.cp36-win_amd64.pyd ，将这两个复制到 .\lib\utils 下（和第6步效果一样。）重构的setup 如下：

from Cython.Build import cythonize
i