Form Mismatch beim Kopieren Gewichte von eigenen trainierten Netz in Caffe

Question

Ich bekomme Form Mismatch Fehler auf FC-Schicht, wenn ich versuche, trainierte Netz in C++ zu laden. My Eingangsschicht in Zug/Prüfphase (unterscheiden sich nur durch die Quelle und die Chargengröße, sind einige params weggelassen):

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

Eingangsschicht zum Bereitstellen von:

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

Fehler:

Cannot copy param 0 weights from layer 'fc4'; shape mismatch. Source param shape is 8 26912 (215296); target param shape is 8 32768 (262144).

Ich fand this Antwort, aber mein eigenes Modell ist geschult und getestet. Könnten Sie etwas Licht auf dieses Thema werfen?

Answer 1

Sie füttern Ihre Netze mit unterschiedlich großen Eingängen: Während des Trainings sind Sie cropping die Eingänge zu 227x277, während Sie während des Tests nicht zuschneiden und die Bilder bei 256x256 verlassen. Dies führt zu dem Fehler, den Sie haben.