最近在学习和浮现CTPN代码,参考的是https://github.com/eragonruan/text-detection-ctpn
bbox_overlaps函数在复现时,原本是打算自己写一个,学习Numpy的相应函数来实现。
但是看到一篇博客说:使用Python复现overlap计算特别慢,而且我所使用的服务器资源也少。
已有cpython现成的代码,编译一下即可。
所以,我就学习了一下cpython使用。ctpn/rcnn源代码里overlap计算就是cpython来实现。
这里我讲一讲如何配置完成吧。
首先,我直接将源码里的bbox.pyx
文件拿过来,学习如何进行编译,参考博客,写了一个setup.py
from distutils.core import setup, Extension
from Cython.Build import cythonize
import numpy as np
setup(ext_modules=cythonize("bbox.pyx"), include_dirs=[np.get_include()])
QA1: 使用python setup.py build
来编译,linux下没问题,但是win就报了错
发现是C++的链接库的问题,就装了个VS 2017,解决了。
QA2: 到了这一步,还没有配置完成,因为我试了试能不能在Jupyter里调用这个函数bbox_overlaps
但找不到,明明bbox.c, bbox.pyx都在目录下,但就是不能import这个模块。
再看看源码里面还有什么不一样呢?
源码的文件夹里有make.sh
,查看其中内容,如下代码。其实就是多做了一步,将so文件(linux)移出来,这样才能找到模块的位置
python setup.py install
mv build/*/*.so ./
rm -rf build/
import成功!
from bbox.bbox import bbox_overlaps