src/caffe/layers/hdf5_output_layer.cu

   1 // Copyright 2014 BVLC and contributors.
   2 /*
   3 Contributors:
   4 - kloudkl@github, 2014.
   5 */
   6
   7 #include <vector>
   8
   9 #include "hdf5.h"
  10 #include "hdf5_hl.h"
  11
  12 #include "caffe/blob.hpp"
  13 #include "caffe/common.hpp"
  14 #include "caffe/layer.hpp"
  15 #include "caffe/util/io.hpp"
  16 #include "caffe/vision_layers.hpp"
  17
  18 namespace caffe {
  19 using std::vector;
  20
  21 template <typename Dtype>
  22 Dtype HDF5OutputLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
  23       vector<Blob<Dtype>*>* top) {
  24   CHECK_GE(bottom.size(), 2);
  25   CHECK_EQ(bottom[0]->num(), bottom[1]->num());
  26   data_blob_.Reshape(bottom[0]->num(), bottom[0]->channels(),
  27                      bottom[0]->height(), bottom[0]->width());
  28   label_blob_.Reshape(bottom[1]->num(), bottom[1]->channels(),
  29                      bottom[1]->height(), bottom[1]->width());
  30   const int data_datum_dim = bottom[0]->count() / bottom[0]->num();
  31   const int label_datum_dim = bottom[1]->count() / bottom[1]->num();
  32
  33   for (int i = 0; i < bottom[0]->num(); ++i) {
  34     CUDA_CHECK(cudaMemcpy(&data_blob_.mutable_cpu_data()[i * data_datum_dim],
  35            &bottom[0]->gpu_data()[i * data_datum_dim],
  36            sizeof(Dtype) * data_datum_dim, cudaMemcpyDeviceToHost));
  37     CUDA_CHECK(cudaMemcpy(&label_blob_.mutable_cpu_data()[i * label_datum_dim],
  38            &bottom[1]->gpu_data()[i * label_datum_dim],
  39            sizeof(Dtype) * label_datum_dim, cudaMemcpyDeviceToHost));
  40   }
  41   SaveBlobs();
  42   return Dtype(0.);
  43 }
  44
  45 template <typename Dtype>
  46 void HDF5OutputLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
  47       const bool propagate_down, vector<Blob<Dtype>*>* bottom) {
  48   return;
  49 }
  50
  51 INSTANTIATE_CLASS(HDF5OutputLayer);
  52
  53 }  // namespace caffe