Channel last to channel first

Tensorflow MaxPooling2D layer takes as input a tensor with the following structure (Batch, Width, Length, Channels), Caffe’s input instead is structured as (Batch, Channels, Width, Length).

I’m working with both frameworks but I would like Caffe to mimics Tensorflow behavior, how can I modify a tensor that produces a specific output on Tensorflow in a way that, when fed to Caffe’s MaxPool, provides the same output?

submitted by /u/sauro97
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *