You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
3.3 KiB
77 lines
3.3 KiB
import unittest
|
|
import numpy as np
|
|
from op_test import OpTest
|
|
|
|
|
|
class TestConv2dOp(OpTest):
|
|
def setUp(self):
|
|
self.op_type = "conv2d"
|
|
batch_size = 2
|
|
input_channels = 3
|
|
input_height = 5
|
|
input_width = 5
|
|
output_channels = 6
|
|
filter_height = 3
|
|
filter_width = 3
|
|
stride = 1
|
|
padding = 0
|
|
groups = 3
|
|
output_height = (input_height - filter_height + 2 * padding
|
|
) / stride + 1
|
|
output_width = (input_width - filter_width + 2 * padding) / stride + 1
|
|
input = np.random.random((batch_size, input_channels, input_height,
|
|
input_width)).astype("float32")
|
|
|
|
filter = np.random.random(
|
|
(output_channels, input_channels / groups, filter_height,
|
|
filter_width)).astype("float32")
|
|
output = np.ndarray(
|
|
(batch_size, output_channels, output_height, output_width))
|
|
|
|
self.inputs = {'Input': input, 'Filter': filter}
|
|
self.attrs = {'strides': [1, 1], 'paddings': [0, 0], 'groups': groups}
|
|
|
|
output_group_channels = output_channels / groups
|
|
input_group_channels = input_channels / groups
|
|
for batchid in xrange(batch_size):
|
|
for group in xrange(groups):
|
|
for outchannelid in range(group * output_group_channels,
|
|
(group + 1) * output_group_channels):
|
|
for rowid in xrange(output_height):
|
|
for colid in xrange(output_width):
|
|
start_h = (rowid * stride) - padding
|
|
start_w = (colid * stride) - padding
|
|
output_value = 0.0
|
|
for inchannelid in range(
|
|
group * input_group_channels,
|
|
(group + 1) * input_group_channels):
|
|
for frowid in xrange(filter_height):
|
|
for fcolid in xrange(filter_width):
|
|
input_value = 0.0
|
|
inrowid = start_h + frowid
|
|
incolid = start_w + fcolid
|
|
if ((inrowid >= 0 and
|
|
inrowid < input_height) and
|
|
(incolid >= 0 and
|
|
incolid < input_width)):
|
|
input_value = input[batchid][
|
|
inchannelid][inrowid][incolid]
|
|
filter_value = filter[outchannelid][
|
|
inchannelid % input_group_channels][
|
|
frowid][fcolid]
|
|
output_value += input_value * filter_value
|
|
output[batchid][outchannelid][rowid][
|
|
colid] = output_value
|
|
|
|
self.outputs = {'Output': output}
|
|
|
|
def test_check_output(self):
|
|
self.check_output()
|
|
|
|
def test_check_grad(self):
|
|
self.check_grad(set(['Input', 'Filter']), 'Output')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|