mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-05 22:19:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added exception when calling forward to specified layer with the new dnn engine

Browse Source
This commit is contained in:
Alexander Lyulkov 2024-10-25 13:00:15 +03:00
parent 9f0c3f5b2b
commit 3a4c88c33e
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/include/opencv2/dnn/dnn.hpp
View File

@ -1018,8 +1018,8 @@ CV__DNN_INLINE_NS_BEGIN
enum EngineType
{
ENGINE_CLASSIC=1, //!< Force use the new dnn engine. The engine does not support non CPU back-ends for now.
ENGINE_NEW=2, //!< Force use the old dnn engine similar to 4.x branch
ENGINE_CLASSIC=1, //!< Force use the old dnn engine similar to 4.x branch
ENGINE_NEW=2, //!< Force use the new dnn engine. The engine does not support non CPU back-ends for now.
ENGINE_AUTO=3 //!< Try to use the new engine and then fall back to the classic version.
};

3
modules/dnn/src/net_impl.cpp
View File

@ -1178,6 +1178,9 @@ void Net::Impl::forward(std::vector<std::vector<Mat>>& outputBlobs,
CV_Assert(!empty());
FPDenormalsIgnoreHintScope fp_denormals_ignore_scope;
if (mainGraph)
CV_Error(Error::StsNotImplemented, "The new dnn engine doesn't support inference until a specified layer. If you want to run the whole model, please don't set the outputName argument in the forward() call. If you want to run the model until a specified layer, please use the old dnn engine");
std::vector<LayerPin> pins;
for (int i = 0; i < outBlobNames.size(); i++)
{