mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.0 KiB
88 lines
2.0 KiB
// This file is part of OpenCV project. |
|
// It is subject to the license terms in the LICENSE file found in the top-level directory |
|
// of this distribution and at http://opencv.org/license.html. |
|
|
|
#include "precomp.hpp" |
|
|
|
#include <sstream> |
|
|
|
#include <opencv2/dnn/layer_reg.private.hpp> |
|
#include <opencv2/dnn/utils/debug_utils.hpp> |
|
#include <opencv2/core/utils/logger.hpp> |
|
|
|
namespace cv { namespace dnn { |
|
CV__DNN_INLINE_NS_BEGIN |
|
|
|
bool DNN_DIAGNOSTICS_RUN = false; |
|
bool DNN_SKIP_REAL_IMPORT = false; |
|
|
|
void enableModelDiagnostics(bool isDiagnosticsMode) |
|
{ |
|
DNN_DIAGNOSTICS_RUN = isDiagnosticsMode; |
|
|
|
if (DNN_DIAGNOSTICS_RUN) |
|
{ |
|
detail::NotImplemented::Register(); |
|
} |
|
else |
|
{ |
|
detail::NotImplemented::unRegister(); |
|
} |
|
} |
|
|
|
void skipModelImport(bool skip) |
|
{ |
|
DNN_SKIP_REAL_IMPORT = skip; |
|
} |
|
|
|
void detail::LayerHandler::addMissing(const std::string& name, const std::string& type) |
|
{ |
|
// If we didn't add it, but can create it, it's custom and not missing. |
|
if (!contains(type) && LayerFactory::isLayerRegistered(type)) |
|
{ |
|
return; |
|
} |
|
|
|
layers[type].insert(name); |
|
} |
|
|
|
bool detail::LayerHandler::contains(const std::string& type) const |
|
{ |
|
return layers.count(type) != 0; |
|
} |
|
|
|
void detail::LayerHandler::printMissing() const |
|
{ |
|
if (layers.empty()) |
|
{ |
|
return; |
|
} |
|
|
|
std::ostringstream ss; |
|
ss << "DNN: Not supported types:\n"; |
|
for (const auto& type_names : layers) |
|
{ |
|
const auto& type = type_names.first; |
|
ss << "Type='" << type << "', affected nodes:\n["; |
|
for (const auto& name : type_names.second) |
|
{ |
|
ss << "'" << name << "', "; |
|
} |
|
ss.seekp(-2, std::ios_base::end); |
|
ss << "]\n"; |
|
} |
|
CV_LOG_ERROR(NULL, ss.str()); |
|
} |
|
|
|
LayerParams detail::LayerHandler::getNotImplementedParams(const std::string& name, const std::string& op) |
|
{ |
|
LayerParams lp; |
|
lp.name = name; |
|
lp.type = "NotImplemented"; |
|
lp.set("type", op); |
|
|
|
return lp; |
|
} |
|
|
|
CV__DNN_INLINE_NS_END |
|
}} // namespace
|
|
|