Caffe预训练模型具有更大的输入图像尺寸

时间:2016-04-22 11:48:43

标签: deep-learning caffe matcaffe

你能否建议我一个训练有素的CNN模型,该模型是在一个支持输入图像尺寸> 400x400像素的更大数据集上训练的?

在我的情况下,我有非常薄的结构,如果我将图像重新缩放到标准尺寸,例如224x224我会松开那些结构。

1 个答案:

答案 0 :(得分:3)

如果没有您的具体应用,很难指出适当的预训练模型。但是,我建议如下:

您不一定需要预先训练好的模型来支持更大的图像尺寸 - 根据您的应用,您需要重新训练最后几层(即完全连接的层)。然而,卷积层(以及汇集层,局部响应归一化层,ReLU层等)与图像大小无关。因此,您可以将卷积层形成预先训练的模型(例如,来自AlexNetreference BVLC model),并根据您的需要调整完全连接的层,并仅重新训练那些。

然而,这里有一些图像尺寸较大的模型(你可以在Caffe Model Zoo中找到所有这些模型)。但是,它可能取决于您的应用是否使用这些模型。您可以查看这些模型的prototxt文件,以查看所需的确切图像大小。

相关问题