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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/memories_sunset/article/details/82117628

错误截图
在训练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

猜你喜欢

转载自blog.csdn.net/memories_sunset/article/details/82117628