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

WIN10+Tensorflow+Faster RCNN训练错误:No module named cython_bbox_第1张图片
在训练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代码情况如下:
WIN10+Tensorflow+Faster RCNN训练错误:No module named cython_bbox_第2张图片
然后重新进行编译,编译成功之后在相应文件夹下,生成一个新的cython_bbox的文件(这里是针对Py36的版本)。我们将这个文件放入对应位置:

WIN10+Tensorflow+Faster RCNN训练错误:No module named cython_bbox_第3张图片

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

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

你可能感兴趣的:(研究生生涯问题杂记)