1、程序版本:Windows10,TensorFlow-gpu==1.5.0,python3.5.6。
2、本文选择的Windows Faster rcnn python版本代码为 https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5。
3、下载预训练的VGG16网络模型:下载地址:http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
网上都是介绍怎么配置Faster-Rcnn怎样训练自己的数据集的,但是如果你想要继续深入研究,找创新点写论文的话,就需要更改模型了。更改之后会遇到以下问题:
问题一:更改模型之后,如加入卷积层或者删除卷积层之后,程序卡在这里不能继续训练。keep_inds = np.append(fg_inds, bg_inds) (Pdb)?
方法:将config.py中的这两句里面的值由0.5,0.1,分别改为0.3,0.0。
tf.app.flags.DEFINE_float('roi_bg_threshold_high', 0.5, "Overlap threshold for a ROI to be