Administrator/opencv-MIRROR
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-09-12 05:11:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

dnn: preserve Cast semantics after ONNX Gather

Browse Source
This commit is contained in:
Yang Guanyuhan
2026-07-28 00:13:40 +08:00
parent 9548c7ea4e
commit 0e36cafcf4
2 changed files with 28 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
modules/dnn/test/test_layers.cpp
View File

@@ -2347,6 +2347,34 @@ TEST(Layer_Size, onnx_0d_scalar)
EXPECT_EQ(outs[0].at<int64_t>(0), 1);
}
TEST(Layer_GatherCast, preserves_float_cast)
{
auto engine_forced = static_cast<cv::dnn::EngineType>(
cv::utils::getConfigurationParameterSizeT("OPENCV_FORCE_DNN_ENGINE", cv::dnn::ENGINE_AUTO));
if (engine_forced == cv::dnn::ENGINE_ORT)
{
applyTestTag(CV_TEST_TAG_DNN_SKIP_PARSER);
return;
}
const std::string modelname = findDataFile("dnn/onnx/models/gather_cast_float.onnx", true);
Net net = readNetFromONNX(modelname, ENGINE_OPENCV);
ASSERT_FALSE(net.empty());
ASSERT_TRUE(net.getMainGraph());
int inputShape[] = {2, 3};
Mat input(2, inputShape, CV_32F, Scalar(0));
net.setInput(input, "input");
std::vector<Mat> outputs;
net.forward(outputs, std::vector<String>{"output"});
ASSERT_EQ(outputs.size(), 1u);
EXPECT_EQ(outputs[0].total(), (size_t)1);
EXPECT_EQ(outputs[0].type(), CV_32F);
EXPECT_FLOAT_EQ(outputs[0].ptr<float>()[0], 2.f);
}
TEST(ConvolutionWinograd, Accuracy)
{
Mat weights({2, 1, 3, 3}, CV_32F);