Advertisement

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

阅读量:

在Windows平台使用Faster RCNN时,可能会遇到Py36编译错误,主要原因是基于Py35的Faster RCNN版本无法在Py36环境中编译。通过修改setup.py中的Extension语句,添加了新的编译选项,并将修改后的代码和编译后的文件放置在正确的位置后,成功解决了问题。修改后的代码可以在PyCharm中使用,使编译成功。

错误截图

在训练Faster RCNN模型时,特别是在Windows操作系统环境下,一些用户可能会遇到类似问题。这种问题的主要原因在于,目前广泛流行的Faster RCNN版本大多基于Py35框架进行构建。当在Py36环境中进行编译时,通常会遇到这些问题。为了优化Faster RCNN项目的编译流程,我们对setup.py文件进行了必要的修改。具体来说,我们在设置文件的第16行新增了Extension项,其参数包括lib.utils.cython_bbox模块的编译配置。该模块的源代码来源于项目根目录下lib/utils/bbox.c和lib/utils/bbox.pyx文件,并且在编译过程中需要包含np.get_include()和 '/lib/utils'等包含目录。此外,我们还为该模块设置了空的extra_compile_args参数。修改完成后,setup.py文件的具体内容如下:

修改后的代码

将整个编译流程分为两个阶段:首先,我们对代码进行重新编译操作,待编译完成后,将在预设的文件夹中生成一个新的cython_bbox文件(该文件专为Python 3.6版本设计)。随后,我们将这个编译好的文件放置于指定位置:

新的文件放置的位置

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

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

全部评论 (0)

还没有任何评论哟~