diff --git a/testdata/dnn/darknet/generate_darknet_models.py b/testdata/dnn/darknet/generate_darknet_models.py index 532daea7b..9a38a3cf6 100644 --- a/testdata/dnn/darknet/generate_darknet_models.py +++ b/testdata/dnn/darknet/generate_darknet_models.py @@ -36,4 +36,5 @@ def genTestData(name, inpShape, outShape): genTestData('route', [1, 4, 3, 6], [1, 2, 2, 3]) genTestData('route_multi', [1, 6, 3, 6], [1, 4, 2, 3]) genTestData('relu', [1, 2, 3, 4], [1, 2, 3, 4]) +genTestData('sam', [1, 2, 3, 4], [1, 4, 3, 4]) diff --git a/testdata/dnn/darknet/sam.cfg b/testdata/dnn/darknet/sam.cfg new file mode 100644 index 000000000..8391986ab --- /dev/null +++ b/testdata/dnn/darknet/sam.cfg @@ -0,0 +1,22 @@ +[net] +batch=1 +channels=2 +height=3 +width=4 + +[convolutional] +filters=4 +size=3 +stride=1 +pad=1 +activation=leaky + +[convolutional] +filters=4 +size=1 +stride=1 +pad=1 +activation=logistic + +[sam] +from=-2 diff --git a/testdata/dnn/darknet/sam.weights b/testdata/dnn/darknet/sam.weights new file mode 100644 index 000000000..ffcf144cf Binary files /dev/null and b/testdata/dnn/darknet/sam.weights differ diff --git a/testdata/dnn/darknet/sam_in.npy b/testdata/dnn/darknet/sam_in.npy new file mode 100644 index 000000000..63b54d2aa Binary files /dev/null and b/testdata/dnn/darknet/sam_in.npy differ diff --git a/testdata/dnn/darknet/sam_out.npy b/testdata/dnn/darknet/sam_out.npy new file mode 100644 index 000000000..486ac814b Binary files /dev/null and b/testdata/dnn/darknet/sam_out.npy differ