reshape DATA + IMAGE_DATA for inputs of varying dimension
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Fri, 17 Oct 2014 05:52:30 +0000 (22:52 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Tue, 17 Feb 2015 04:10:17 +0000 (20:10 -0800)
commit82551adbf8c77a06fa7e8b2508e1fb24ff1b5b31
tree57f4bb0a8dc79a0ab63376322f0731d25cee8343
parent1344d1bc51f781b593a52fa71ecd59a23dc031fe
reshape DATA + IMAGE_DATA for inputs of varying dimension

To feed inputs of varying dimension, the `DATA` and `IMAGE_DATA` layer
reshapes its prefetch and top blobs when the batch size is 1.

The `BasePrefetchingDataLayer` always reshapes on forward.
src/caffe/layers/base_data_layer.cpp
src/caffe/layers/base_data_layer.cu
src/caffe/layers/data_layer.cpp
src/caffe/layers/image_data_layer.cpp