1 // Copyright 2013 Yangqing Jia
3 #include <cstring>
4 #include <cuda_runtime.h>
6 #include "gtest/gtest.h"
7 #include "caffe/blob.hpp"
8 #include "caffe/common.hpp"
9 #include "caffe/filler.hpp"
10 #include "caffe/vision_layers.hpp"
11 #include "caffe/test/test_gradient_check_util.hpp"
13 #include "caffe/test/test_caffe_main.hpp"
15 namespace caffe {
17 extern cudaDeviceProp CAFFE_TEST_CUDA_PROP;
19 template <typename Dtype>
20 class Im2colLayerTest : public ::testing::Test {
21 protected:
22 Im2colLayerTest()
23 : blob_bottom_(new Blob<Dtype>(2, 3, 6, 5)),
24 blob_top_(new Blob<Dtype>()) {
25 // fill the values
26 FillerParameter filler_param;
27 GaussianFiller<Dtype> filler(filler_param);
28 filler.Fill(this->blob_bottom_);
29 blob_bottom_vec_.push_back(blob_bottom_);
30 blob_top_vec_.push_back(blob_top_);
31 };
32 virtual ~Im2colLayerTest() { delete blob_bottom_; delete blob_top_; }
33 Blob<Dtype>* const blob_bottom_;
34 Blob<Dtype>* const blob_top_;
35 vector<Blob<Dtype>*> blob_bottom_vec_;
36 vector<Blob<Dtype>*> blob_top_vec_;
37 };
39 typedef ::testing::Types<float, double> Dtypes;
40 TYPED_TEST_CASE(Im2colLayerTest, Dtypes);
42 TYPED_TEST(Im2colLayerTest, TestSetup) {
43 LayerParameter layer_param;
44 layer_param.set_kernelsize(3);
45 layer_param.set_stride(2);
46 Im2colLayer<TypeParam> layer(layer_param);
47 layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
48 EXPECT_EQ(this->blob_top_->num(), 2);
49 EXPECT_EQ(this->blob_top_->channels(), 27);
50 EXPECT_EQ(this->blob_top_->height(), 2);
51 EXPECT_EQ(this->blob_top_->width(), 2);
52 }
54 TYPED_TEST(Im2colLayerTest, TestCPU) {
55 LayerParameter layer_param;
56 layer_param.set_kernelsize(3);
57 layer_param.set_stride(2);
58 Im2colLayer<TypeParam> layer(layer_param);
59 Caffe::set_mode(Caffe::CPU);
60 layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
61 layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_));
62 // We are lazy and will only check the top left block
63 for (int c = 0; c < 27; ++c) {
64 EXPECT_EQ(this->blob_top_->data_at(0, c, 0, 0),
65 this->blob_bottom_->data_at(0, (c / 9), (c / 3) % 3, c % 3));
66 }
67 }
69 TYPED_TEST(Im2colLayerTest, TestGPU) {
70 LayerParameter layer_param;
71 layer_param.set_kernelsize(3);
72 layer_param.set_stride(2);
73 Im2colLayer<TypeParam> layer(layer_param);
74 Caffe::set_mode(Caffe::GPU);
75 layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
76 layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_));
77 // We are lazy and will only check the top left block
78 for (int c = 0; c < 27; ++c) {
79 EXPECT_EQ(this->blob_bottom_->data_at(0, (c / 9), (c / 3) % 3, c % 3),
80 this->blob_top_->data_at(0, c, 0, 0));
81 }
82 }
84 TYPED_TEST(Im2colLayerTest, TestCPUGradient) {
85 LayerParameter layer_param;
86 layer_param.set_kernelsize(3);
87 layer_param.set_stride(2);
88 Caffe::set_mode(Caffe::CPU);
89 Im2colLayer<TypeParam> layer(layer_param);
90 GradientChecker<TypeParam> checker(1e-2, 1e-2);
91 checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
92 }
94 TYPED_TEST(Im2colLayerTest, TestGPUGradient) {
95 LayerParameter layer_param;
96 layer_param.set_kernelsize(3);
97 layer_param.set_stride(2);
98 Caffe::set_mode(Caffe::GPU);
99 Im2colLayer<TypeParam> layer(layer_param);
100 GradientChecker<TypeParam> checker(1e-2, 1e-2);
101 checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
102 }
105 }