CPU下配置Faster_RCNN以及训练自己的数据集
CPU下Faster-RCNN环境配置以及训练自己的数据集
环境
Ubuntu16.04
python 2.7
tensorflow1.7.0
下载源码
git clone https://github.com/endernewton/tf-faster-rcnn.git
修改配置
1修改lib/setup.py
将__C.USE——GPU——NMSTRUE 修改为 **C.USE——GPU——NMSTRUE**
2修改lib/model/nms_wrapper.py
按照下图注释掉两处
原创
2022-04-04 17:08:21 ·
1734 阅读 ·
0 评论