public void Test_AvgPool2D_Null_Input()
{
Data2D data = null;
AvgPool2DLayer pool = new AvgPool2DLayer(1, 1, 1, 1, 2, 2);
pool.SetInput(data);
}
public void Test_AvgPool2D_DifferentData_Input()
{
DataArray data = new DataArray(5);
AvgPool2DLayer pool = new AvgPool2DLayer(1, 1, 1, 1, 2, 2);
pool.SetInput(data);
}
public void Test_AvgPool2D_Execute()
{
// Initialize data.
Data2D data = new Data2D(3, 3, 2, 1);
data[0, 0, 0, 0] = 2;
data[1, 0, 0, 0] = 1;
data[2, 0, 0, 0] = 1;
data[0, 1, 0, 0] = 1;
data[1, 1, 0, 0] = 4;
data[2, 1, 0, 0] = 2;
data[0, 2, 0, 0] = 1;
data[1, 2, 0, 0] = 2;
data[2, 2, 0, 0] = 0;
data[0, 0, 1, 0] = 3;
data[1, 0, 1, 0] = 6;
data[2, 0, 1, 0] = 8;
data[0, 1, 1, 0] = 5;
data[1, 1, 1, 0] = 2;
data[2, 1, 1, 0] = 4;
data[0, 2, 1, 0] = 1;
data[1, 2, 1, 0] = 4;
data[2, 2, 1, 0] = 2;
AvgPool2DLayer pool = new AvgPool2DLayer(0, 0, 1, 1, 2, 2);
pool.SetInput(data);
pool.Execute();
Data2D output = pool.GetOutput() as Data2D;
// Checking sizes
Dimension dim = output.GetDimension();
Assert.AreEqual(dim.b, 1);
Assert.AreEqual(dim.c, 2);
Assert.AreEqual(dim.h, 2);
Assert.AreEqual(dim.w, 2);
// Checking calculation
Assert.AreEqual(output[0, 0, 0, 0], 2.0, 0.0000001);
Assert.AreEqual(output[1, 0, 0, 0], 2.0, 0.0000001);
Assert.AreEqual(output[0, 1, 0, 0], 2.0, 0.0000001);
Assert.AreEqual(output[1, 1, 0, 0], 2.0, 0.0000001);
Assert.AreEqual(output[0, 0, 1, 0], 4.0, 0.0000001);
Assert.AreEqual(output[1, 0, 1, 0], 5.0, 0.0000001);
Assert.AreEqual(output[0, 1, 1, 0], 3.0, 0.0000001);
Assert.AreEqual(output[1, 1, 1, 0], 3.0, 0.0000001);
}
