WIN10+Tensorflow+Faster RCNN训练错误:No module named cython_bbox

在Windows 10上训练Faster RCNN时,遇到'No module named cython_bbox'错误。此问题源于多数Faster RCNN版本兼容Python 3.5,而非3.6。解决方案是修改setup.py文件,添加Extension以编译适用于Py36的cython_bbox模块。完成修改并重新编译后,将生成的cython_bbox文件放入正确目录,使用PyCharm打开Train.py即可解决问题。
摘要由CSDN通过智能技术生成

错误截图
在训练Faster RCNN的时候,在win平台下,有一些倒霉的小伙伴或许会出现如图的错误。这个错误的主要原因是,目前网上流行的Faster RCNN版本大多都是基于Py35的。当你使用Py36的环境去编译,那么就会出现这个问题。
我们需要对Faster RCNN项目当中的setup.py进行一点点修改。我们在程序的16行加上Extension( “lib.utils.cython_bbox”, sources=[“../../../lib/utils/bbox.c”,”../../../lib/utils/bbox.pyx”], include_dirs = [np.get_include(), ‘/lib/utils’], extra_compile_args=[], )
加完这些代码之后的Setup.py代码情况如下:
修改后的代码
然后重新进行编译,编译成功之后在相应文件夹下,生成一个新的cython_bbox的文件(这里是针对Py36的版本)。我们将这个文件放入对应位置:

新的文件放置的位置

之后使用PyCharm打开Train.py就可以解决这个问题啦!

注:给出针对的代码的github地址,确保不要弄错哟
针对WIn平台的Faster RCNN

评论 85
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值