Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add padding to max pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-26 19:03:10 +02:00
parent 4a67b708f0
commit 94a16b4352
7 changed files with 78 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/layers/test_max_pooling.cu
View File

@@ -11,6 +11,7 @@ class MaxPoolingLayerTest : public ::testing::Test {
int nChannels;
dim2d poolingSize;
dim2d stride;
dim2d padding;
std::vector<float> input;
std::vector<float> expected;
@@ -35,7 +36,7 @@ class MaxPoolingLayerTest : public ::testing::Test {
cudaError_t cudaStatus;
maxPoolingLayer = new CUDANet::Layers::MaxPooling2d(
inputSize, nChannels, poolingSize, stride,
inputSize, nChannels, poolingSize, stride, padding,
CUDANet::Layers::ActivationType::NONE
);
@@ -71,6 +72,7 @@ TEST_F(MaxPoolingLayerTest, MaxPoolForwardTest) {
nChannels = 2;
poolingSize = {2, 2};
stride = {2, 2};
padding = {0, 0};
input = {
// clang-format off
@@ -97,6 +99,7 @@ TEST_F(MaxPoolingLayerTest, MaxPoolForwardNonSquareInputTest) {
nChannels = 2;
poolingSize = {2, 2};
stride = {2, 2};
padding = {0, 0};
input = {// Channel 0
0.573f, 0.619f, 0.732f, 0.055f, 0.123f, 0.234f, 0.243f, 0.316f,
@@ -118,6 +121,7 @@ TEST_F(MaxPoolingLayerTest, MaxPoolForwardNonSquarePoolSizeTest) {
nChannels = 2;
poolingSize = {2, 3}; // Non-square pooling size
stride = {2, 2};
padding = {0, 0};
input = {
// clang-format off
@@ -145,6 +149,7 @@ TEST_F(MaxPoolingLayerTest, MaxPoolForwardNonSquareStrideTest) {
nChannels = 2;
poolingSize = {2, 2};
stride = {1, 2}; // Non-square stride
padding = {0, 0};
input = {
// clang-format off
@@ -165,4 +170,32 @@ TEST_F(MaxPoolingLayerTest, MaxPoolForwardNonSquareStrideTest) {
runTest();
}
TEST_F(MaxPoolingLayerTest, MaxPoolForwardNonSquarePaddingTest) {
inputSize = {4, 4};
nChannels = 2;
poolingSize = {2, 2};
stride = {2, 2}; // Non-square stride
padding = {0, 1};
input = {
// clang-format off
// Channel 0
0.573f, 0.619f, 0.732f, 0.055f,
0.243f, 0.316f, 0.573f, 0.619f,
0.712f, 0.055f, 0.243f, 0.316f,
0.573f, 0.619f, 0.742f, 0.055f,
// Channel 1
0.473f, 0.919f, 0.107f, 0.073f,
0.073f, 0.362f, 0.973f, 0.059f,
0.473f, 0.455f, 0.283f, 0.416f,
0.532f, 0.819f, 0.732f, 0.850f
// clang-format on
};
expected = {0.573f, 0.732f, 0.619f, 0.712f, 0.742f, 0.316f, 0.473f, 0.973f, 0.073f, 0.532f, 0.819f, 0.85f};
runTest();
}

2
test/model/test_model.cu
View File

@@ -45,7 +45,7 @@ class ModelTest : public ::testing::Test {
CUDANet::Layers::MaxPooling2d *maxpool2d =
new CUDANet::Layers::MaxPooling2d(
poolingInput, numFilters, poolingSize,
poolingStride, CUDANet::Layers::ActivationType::RELU
poolingStride, {0, 0}, CUDANet::Layers::ActivationType::RELU
);
model->addLayer("maxpool1", maxpool2d);