namespace caffe {
template <typename Dtype>
-Dtype ImagesLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
+Dtype ImageDataLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top) {
// First, join the thread
CHECK(!pthread_join(thread_, NULL)) << "Pthread joining failed.";
prefetch_label_->cpu_data(), sizeof(Dtype) * prefetch_label_->count(),
cudaMemcpyHostToDevice));
// Start a new prefetch thread
- CHECK(!pthread_create(&thread_, NULL, ImagesLayerPrefetch<Dtype>,
+ CHECK(!pthread_create(&thread_, NULL, ImageDataLayerPrefetch<Dtype>,
reinterpret_cast<void*>(this))) << "Pthread execution failed.";
return Dtype(0.);
}
-INSTANTIATE_CLASS(ImagesLayer);
+INSTANTIATE_CLASS(ImageDataLayer);
} // namespace caffe
EUCLIDEAN_LOSS = 7;
FLATTEN = 8;
HDF5_DATA = 9;
- IM2COL = 10;
- IMAGE_DATA = 11;
- INFOGAIN_LOSS = 12;
- INNER_PRODUCT = 13;
- LRN = 14;
- MULTINOMIAL_LOGISTIC_LOSS = 15;
- POOLING = 16;
- RELU = 17;
- SIGMOID = 18;
- SOFTMAX = 19;
- SOFTMAX_LOSS = 20;
- SPLIT = 21;
- TANH = 22;
- WINDOW_DATA = 23;
+ HDF5_OUTPUT = 10;
+ IM2COL = 11;
+ IMAGE_DATA = 12;
+ INFOGAIN_LOSS = 13;
+ INNER_PRODUCT = 14;
+ LRN = 15;
+ MULTINOMIAL_LOGISTIC_LOSS = 16;
+ POOLING = 17;
+ RELU = 18;
+ SIGMOID = 19;
+ SOFTMAX = 20;
+ SOFTMAX_LOSS = 21;
+ SPLIT = 22;
+ TANH = 23;
+ WINDOW_DATA = 24;
}
optional LayerType type = 2; // the layer type from the enum above
optional DataParameter data_param = 10;
optional DropoutParameter dropout_param = 11;
optional HDF5DataParameter hdf5_data_param = 12;
- optional ImageDataParameter image_data_param = 13;
- optional InfogainLossParameter infogain_loss_param = 14;
- optional InnerProductParameter inner_product_param = 15;
- optional LRNParameter lrn_param = 16;
- optional PoolingParameter pooling_param = 17;
- optional WindowDataParameter window_data_param = 18;
+ optional HDF5OutputParameter hdf5_output_param = 13;
+ optional ImageDataParameter image_data_param = 14;
+ optional InfogainLossParameter infogain_loss_param = 15;
+ optional InnerProductParameter inner_product_param = 16;
+ optional LRNParameter lrn_param = 17;
+ optional PoolingParameter pooling_param = 18;
+ optional WindowDataParameter window_data_param = 19;
}
// Message that stores parameters used by ConcatLayer
optional uint32 batch_size = 2;
}
+// Message that stores parameters used by HDF5OutputLayer
+message HDF5OutputParameter {
+ optional string file_name = 1;
+}
+
// Message that stores parameters used by ImageDataLayer
message ImageDataParameter {
// Specify the data source.
// square: the tightest square around the window is cropped
optional string crop_mode = 11 [default = "warp"];
}
-
-message HDF5OutputParameter {
- optional string file_name = 1;
-}