Merge pull request #842 from l-bat:lb/onnx_normalize

opencv · Feb 3, 2021 · 960de6c · 960de6c
2 parents 61ea2e2 + 00d6a27
commit 960de6c
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/testdata/dnn/onnx/data/input_normalize_fusion.npy b/testdata/dnn/onnx/data/input_normalize_fusion.npy
diff --git a/testdata/dnn/onnx/data/output_normalize_fusion.npy b/testdata/dnn/onnx/data/output_normalize_fusion.npy
diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
@@ -1245,3 +1245,16 @@ def forward(self, x):
 x = Variable(torch.randn([1, 1, 3, 4]))
 model = PadCalculation()
 save_data_and_model("calc_pads", x, model, version=11)
+
+class NormalizeFusion(nn.Module):
+    def forward(self, x):
+        mul = x * x
+        sum = torch.sum(mul, dim=(1), keepdim=True)
+        maximum = torch.clamp(sum, min=1e-8)
+        sqrt = torch.sqrt(maximum)
+        reciprocal = torch.reciprocal(sqrt)
+        return x * reciprocal
+
+x = Variable(torch.randn([2, 3]))
+model = NormalizeFusion()
+save_data_and_model("normalize_fusion", x, model)
diff --git a/testdata/dnn/onnx/models/normalize_fusion.onnx b/testdata/dnn/onnx/models/normalize_fusion.onnx