如果我想在 caffe 模型上测试期间更改输入维度,是否需要进行重新训练



我最近被分配了一个使用Caffe模型的项目。到目前为止,我设法运行测试代码并获得了良好的结果。但由于它只输出小分辨率图像(128x128(,我被要求获得更高的分辨率,例如512x512。由于我刚刚开始在这个项目中学习咖啡,我仍然在理解咖啡的流动过程时遇到一些问题,我偶然发现了这个问题。

这是deploy.prototxt,这里是train.prototxt

那么实际上是否可以只修改 deploy.prototxt,例如将输入维度和一些输出数量更改为我想要的分辨率,而无需重新训练新模型,或者我应该创建一个带有额外层的新模型以实现更高分辨率的下采样和上采样并进行重新训练?如果有人能给我一些解释,那就太好了。感谢您抽出时间阅读本文。

网络将针对 128x128 进行优化。因此,您无法将输入放入 512x512 或查看输出。

若要用作 512x512 大小的输入,必须获取 512x512 大小的数据集并重新训练它。 获取数据集将既困难又耗时。

因此,还有一种方法可以使用Python更改输出分辨率。

最新更新