形狀不匹配，同時從咖啡館自己訓練的網絡復制重量

Question

當我嘗試在C ++中加載訓練好的網絡時，我在FC層上出現了形狀不匹配錯誤。 我在TRAIN / TEST階段的輸入層（僅根據源和批量大小不同，省略了一些參數）：

layer {
    type: "ImageData"
    image_data_param {
        batch_size: 8
        new_height: 256
        new_width: 256
    }
    transform_param {
        crop_size: 227
    }
}

部署輸入層：

layer {
    type: "Input"
    input_param { shape: { dim: 1 dim: 3 dim: 256 dim: 256 }}
}

錯誤：

無法從圖層'fc4'復制參數0權重; 形狀不匹配。 源param形狀是8 26912（215296）; 目標參數形狀是8 32768（262144）。

我找到了這個答案，但我自己的模型經過培訓和測試。 你能否對此有所了解？

Answer 1

您正在為您的網絡提供不同大小的輸入：在訓練期間，您將輸入裁剪為227x277，而在測試期間，您不裁剪並將圖像保留為256x256。 這會導致你得到的錯誤。