mirror of
https://github.com/opencv/opencv.git
synced 2026-07-31 00:03:03 +04:00
Merge pull request #19693 from LupusSanctus:onnx_diagnostic
ONNX diagnostic tool * Final * Add forgotten Normalize layer to the set of supported types * ONNX diagnostic tool corrections * Fixed CI test warnings * Added code minor corrections Co-authored-by: Sergey Slashchinin <sergei.slashchinin@xperience.ai>
This commit is contained in:
@@ -100,6 +100,18 @@ CV__DNN_INLINE_NS_BEGIN
|
||||
CV_EXPORTS std::vector< std::pair<Backend, Target> > getAvailableBackends();
|
||||
CV_EXPORTS_W std::vector<Target> getAvailableTargets(dnn::Backend be);
|
||||
|
||||
/**
|
||||
* @brief Enables detailed logging of the DNN model loading with CV DNN API.
|
||||
* @param[in] isDiagnosticsMode Indicates whether diagnostic mode should be set.
|
||||
*
|
||||
* Diagnostic mode provides detailed logging of the model loading stage to explore
|
||||
* potential problems (ex.: not implemented layer type).
|
||||
*
|
||||
* @note In diagnostic mode series of assertions will be skipped, it can lead to the
|
||||
* expected application crash.
|
||||
*/
|
||||
CV_EXPORTS void enableModelDiagnostics(bool isDiagnosticsMode);
|
||||
|
||||
/** @brief This class provides all data needed to initialize layer.
|
||||
*
|
||||
* It includes dictionary with scalar params (which can be read by using Dict interface),
|
||||
|
||||
@@ -0,0 +1,23 @@
|
||||
// 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.
|
||||
|
||||
#ifndef OPENCV_DNN_LAYER_REG_HPP
|
||||
#define OPENCV_DNN_LAYER_REG_HPP
|
||||
#include <opencv2/dnn.hpp>
|
||||
|
||||
namespace cv {
|
||||
namespace dnn {
|
||||
CV__DNN_INLINE_NS_BEGIN
|
||||
//! @addtogroup dnn
|
||||
//! @{
|
||||
|
||||
//! Register layer types of DNN model.
|
||||
typedef std::map<std::string, std::vector<LayerFactory::Constructor> > LayerFactory_Impl;
|
||||
LayerFactory_Impl& getLayerFactoryImpl();
|
||||
|
||||
//! @}
|
||||
CV__DNN_INLINE_NS_END
|
||||
}
|
||||
}
|
||||
#endif
|
||||
Reference in New Issue
Block a user