书接上文,在经过训练后,模型会被保存在default中,需要把它移动到output文件夹下,具体路径如图所示
这里要注意,程序中原本的训练最大迭代次数为40000,而demo.py中读取的是迭代次数为70000的结果,我们修改后执行demo.py,当然也可以修改最大迭代次数,不过这个要在训练之前修改。
具体修改文件位置是:
修改最大迭代次数:lib/config/config.py中的tf.app.flags.DEFINE_integer('max_iters', 70000, "Max iteration");70000代表最大迭代次数;
或者修改demo.py文件中NETS = {'vgg16': ('vgg16_faster_rcnn_iter_40000.ckpt',), 'res101': ('res101_faster_rcnn_iter_110000.ckpt',)};这里40000为要读取的模型文件对应的迭代轮数。
最后回到最初的目录,执行python demo.py就可以看到faster-rcnn的结果了,下边上40000迭代的模型的几张效果图,预测还是很准的。