win10+tensorflow faster-RCNN 训练自己的数据集

最新推荐文章于 2025-10-26 00:15:56 发布

原创

最新推荐文章于 2025-10-26 00:15:56 发布 · 1w 阅读

82 ·

CC 4.0 BY-SA版权

本文档详细介绍了在Windows 10上使用Tensorflow 1.2.1和Anaconda3训练Faster R-CNN模型的全过程，包括环境配置、源码修改、数据集准备、训练过程等步骤，特别强调了在非Python3.5环境下编译源码的必要性以及数据集标注和格式转换的注意事项。

首先，感谢博客上各路大佬的无私奉献！但是也不得不吐槽下，大佬些写博客的时候能尽量写的对小白友好一点吗？期间遇到各种坑，说多了都是泪啊！话不多说，上正题！

环境：win10+anaconda3+tensorflow1.2.1+python3.6，CPU版本。

源码地址：https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5

一下载源码以及源码环境准备

肯定是下载源码，配置好相应的环境！anaconda prompt下激活tensorflow环境，包括但不限于安装python包（cython，python-opencv，easydict），反正就是提示缺少什么，就install什么就对了！
GitHub上源码提示有“Checkout this branch”这一步，但我没懂什么意思！没照做，但是也训练成功了，有懂得老哥帮忙解释一下！
setup.py文件修改

在第16行加入以下代码，（别问我为什么，我只是代码的搬运工）

    Extension(
            'lib.utils.cython_bbox', 
            sources=['E:/python/tensorflow/Faster-RCNN-TensorFlow-Python3.5-master/lib/utils/bbox.c','E:/python/tensorflow/Faster-RCNN-TensorFlow-Python3.5-master/lib/utils/bbox.pyx'],
            include_dirs = [np.get_include(