diff --git a/testdata/dnn/onnx/data/README.md b/testdata/dnn/onnx/data/README.md new file mode 100644 index 0000000000000000000000000000000000000000..5c2cf7ddab10ef614c9c7fd75315d0e0795c2aec --- /dev/null +++ b/testdata/dnn/onnx/data/README.md @@ -0,0 +1,22 @@ +### OpenCV: Open Source Computer Vision Library + +This repository contains extra data for the OpenCV library. + +#### Resources +* Homepage: http://opencv.org +* Docs: http://docs.opencv.org +* Q&A forum: https://forum.opencv.org + * previous forum (read only): http://answers.opencv.org +* Issue tracking: https://github.com/opencv/opencv/issues + +#### Contributing + +Please read before starting work on a pull request: https://github.com/opencv/opencv/wiki/How_to_contribute + +Summary of guidelines: + +* One pull request per issue; +* Choose the right base branch; +* Include tests and documentation; +* Clean up "oops" commits before submitting; +* Follow the coding style guide. diff --git a/testdata/dnn/onnx/data/input_quantized_gemm.npy b/testdata/dnn/onnx/data/input_quantized_gemm.npy new file mode 100644 index 0000000000000000000000000000000000000000..fe19a11597d98f220cbeee0a1a224ec84782da19 Binary files /dev/null and b/testdata/dnn/onnx/data/input_quantized_gemm.npy differ diff --git a/testdata/dnn/onnx/data/input_squeeze_axes_op13.npy b/testdata/dnn/onnx/data/input_squeeze_axes_op13.npy new file mode 100644 index 0000000000000000000000000000000000000000..93645684dce2c4a5352cee9ed666d41821f03c5f Binary files /dev/null and b/testdata/dnn/onnx/data/input_squeeze_axes_op13.npy differ diff --git a/testdata/dnn/onnx/data/output_quantized_gemm.npy b/testdata/dnn/onnx/data/output_quantized_gemm.npy new file mode 100644 index 0000000000000000000000000000000000000000..9ea62780a2161692de31f85f0e1dbd3c84e4a6b1 Binary files /dev/null and b/testdata/dnn/onnx/data/output_quantized_gemm.npy differ diff --git a/testdata/dnn/onnx/data/output_squeeze_axes_op13.npy b/testdata/dnn/onnx/data/output_squeeze_axes_op13.npy new file mode 100644 index 0000000000000000000000000000000000000000..ba28e14891fb3600c1dec4ab49ce0a14330ef883 Binary files /dev/null and b/testdata/dnn/onnx/data/output_squeeze_axes_op13.npy differ diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index 8fa2334643513af9331b247cfbf2471259d73de8..369432533403415957243be9c274af99f65ee92b 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -774,6 +774,7 @@ input = Variable(torch.randn(3, 1, 2, 4)) model = Squeeze() model.eval() save_data_and_model("squeeze", input, model) +save_data_and_model("squeeze_axes_op13", input, model, version=13) class Div(nn.Module): diff --git a/testdata/dnn/onnx/generate_quantized_onnx_models.py b/testdata/dnn/onnx/generate_quantized_onnx_models.py index 85c934180a24cfa4a3a3548ca0a8ef144100bfc6..33a90190c9644ede57f26317e15f387dff5620e7 100644 --- a/testdata/dnn/onnx/generate_quantized_onnx_models.py +++ b/testdata/dnn/onnx/generate_quantized_onnx_models.py @@ -269,4 +269,13 @@ model = nn.Sequential( nn.Linear(84, 10) ) input = Variable(torch.randn(1, 3, 32, 32)) -quantize_and_save_model("quantized_constant", input, model, wt_type="int8", per_channel=True) \ No newline at end of file +quantize_and_save_model("quantized_constant", input, model, wt_type="int8", per_channel=True) + +class Gemm(nn.Module): + def forward(self, x): + mat1 =torch.ones(3, 3) + return torch.mm(x, mat1) + +input = Variable(torch.randn(1, 3)) +model = Gemm() +quantize_and_save_model("quantized_gemm", input, model, act_type="int8", wt_type="int8", per_channel=False) \ No newline at end of file diff --git a/testdata/dnn/onnx/models/quantized_gemm.onnx b/testdata/dnn/onnx/models/quantized_gemm.onnx new file mode 100644 index 0000000000000000000000000000000000000000..052c9afa2b86efcaae44fa8bdbe4d39006ca268f Binary files /dev/null and b/testdata/dnn/onnx/models/quantized_gemm.onnx differ diff --git a/testdata/dnn/onnx/models/squeeze_axes_op13.onnx b/testdata/dnn/onnx/models/squeeze_axes_op13.onnx new file mode 100644 index 0000000000000000000000000000000000000000..88340e794ac5045fef77c940a55d37688f514653 Binary files /dev/null and b/testdata/dnn/onnx/models/squeeze_axes_op13.onnx differ