在caffe上使用VGG16的自定义数据集上py-faster-rcnn的管道

时间:2016-07-07 20:50:17

标签: machine-learning computer-vision deep-learning caffe

我想使用在imagenet上训练的现有VGG16模型,并使用自定义数据集对其进行微调,以用于我所需的其他类。

我在哪里可以找到相同的caffemodel,train_val.prototxt和solver.prototxt?

要使用自定义数据集对其进行微调,其过程与

相同

Fine Tuning of GoogLeNet Model

A guide to convert_imageset.cpp

但是,我想使用新训练的VGG16模型权重在自定义数据集上训练更快的RCNN(py-faster-rcnn)https://github.com/rbgirshick/py-faster-rcnn。 为了在自定义数据集上训练更快的RCNN,我计划遵循此处给出的步骤http://sgsai.blogspot.com/2016/02/training-faster-r-cnn-on-custom-dataset.html

先前完成的VGG16微调产生的caffemodel是否可以在这里工作或者需要进行一些调整?

0 个答案:

没有答案