2017-12-16 9 views
1

Was sind die Unterschiede zwischen der Deconvolution-Schicht in Caffe und Tensorflow? In Tensoroflow gibt es zwei Padding-Modi: "SAME" und "VALID", was ist gleich Padding-Modus, der in Caffe verwendet wurde?Dekonvolution in Tensorflow vs. Caffe

Zum Beispiel in Tensorflow, die Entfaltungs Schicht ist:

decv = slim.convolution2d_transpose(in_layer, num_outputs=256, kernel_size=[8, 8], stride=4, padding='VALID', scope='decv') 

Und in Caffe, es ist:

layer { 
    name: "decv" 
    type: "Deconvolution" 
    bottom: "some_layer" 
    top: "some_layer" 
    param { lr_mult: 1 decay_mult: 1.0 } 
    param { lr_mult: 2 decay_mult: 0} 
    convolution_param { 
    num_output: 256 
    pad: 1 stride: 4 kernel_size: 8 
    group: 256 
    weight_filler { type: "bilinear" } 
    } 
} 

Ich kann die gleiche Upsampling Karte nicht erreichen, wenn die Übertragung des Codes von Caffe zu Tensorflow.

Antwort

0

'VALID' bedeutet keine Polsterung, aber Ihre Caffe Schicht wird mit Kissen gepolstert: 1.

Try 'SAME' padding