diff --git a/modules/dnn/src/tflite/tflite_importer.cpp b/modules/dnn/src/tflite/tflite_importer.cpp index a23bff2545..8850cd9ad2 100644 --- a/modules/dnn/src/tflite/tflite_importer.cpp +++ b/modules/dnn/src/tflite/tflite_importer.cpp @@ -300,6 +300,10 @@ void TFLiteImporter::addLayer(LayerParams& layerParams, const Operator& op) { Mat blob = allTensors[idx]; layerParams.blobs.push_back(blob.u ? blob : blob.clone()); // some tensors are owned by OpenCV } + } else { + for (auto& blob : layerParams.blobs) { + CV_Assert(blob.u); + } } int dtype = CV_32F; @@ -830,9 +834,6 @@ void TFLiteImporter::parseFullyConnected(const Operator& op, const std::string& auto options = op.builtin_options_as_FullyConnectedOptions(); CV_Assert(options); - int idx = op.inputs()->Get(1); - Mat weights = allTensors[idx]; - layerParams.blobs.resize(1, weights); layerParams.set("transB", true); layerParams.set("constB", true); addLayer(layerParams, op); diff --git a/modules/dnn/test/test_tflite_importer.cpp b/modules/dnn/test/test_tflite_importer.cpp index 26f2c373b8..291d1f50d2 100644 --- a/modules/dnn/test/test_tflite_importer.cpp +++ b/modules/dnn/test/test_tflite_importer.cpp @@ -240,8 +240,6 @@ TEST_P(Test_TFLite, split) { } TEST_P(Test_TFLite, fully_connected) { - if (backend == DNN_BACKEND_CUDA) - applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA, CV_TEST_TAG_DNN_SKIP_CUDA_FP16); if (backend == DNN_BACKEND_VKCOM) applyTestTag(CV_TEST_TAG_DNN_SKIP_VULKAN); testLayer("fully_connected");