CTPN/Faster-RCNN中使用cpython对overlap计算加速

最近在学习和浮现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
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值