TYPED_TEST(HardSpatialTransformerLayerTest, TestGradient) {
typedef typename TypeParam::Dtype Dtype;
bool IS_VALID_CUDA = false;
#ifndef CPU_ONLY
IS_VALID_CUDA = CAFFE_TEST_CUDA_PROP.major >= 2;
#endif
if (Caffe::mode() == Caffe::CPU ||
sizeof(Dtype) == 4 || IS_VALID_CUDA) {
// reshape theta to have full 6 dimension
vector<int> shape_theta(2);
shape_theta[0] = 2;
shape_theta[1] = 6;
this->blob_theta_->Reshape(shape_theta);
// fill random variables for theta
FillerParameter filler_param;
GaussianFiller<Dtype> filler(filler_param);
filler.Fill(this->blob_theta_);
// set layer_param
LayerParameter layer_param;
SpatialTransformerParameter *st_param = layer_param.mutable_st_param();
st_param->set_output_h(7);
st_param->set_output_w(7);
// begin to check
SpatialTransformerLayer<Dtype> layer(layer_param);
GradientChecker<Dtype> checker(1e-6, 1e-6);
checker.CheckGradientExhaustive(&layer, this->blob_bottom_vec_,
this->blob_top_vec_);
} else {
LOG(ERROR) << "Skipping test due to old architecture.";
}
}
HardSpatialTransformerLayerTest()
: blob_U_(new Blob<Dtype>(2, 3, 10, 10)),
blob_theta_(new Blob<Dtype>(5, 2, 3, 1)),
blob_V_(new Blob<Dtype>(2, 3, 10, 10)) {
FillerParameter filler_param;
GaussianFiller<Dtype> filler(filler_param);
filler.Fill(this->blob_U_);
filler.Fill(this->blob_theta_);
vector<int> shape_theta(2);
shape_theta[0] = 2; shape_theta[1] = 6;
blob_theta_->Reshape(shape_theta);
blob_bottom_vec_.push_back(blob_U_);
blob_bottom_vec_.push_back(blob_theta_);
blob_top_vec_.push_back(blob_V_);
}
