mirror of
https://github.com/opencv/opencv.git
synced 2025-06-11 11:45:30 +08:00
6801dd043d
fix ONNXImporter diagnostic mode layer registration issue * fix layer registration, thread unsafe access and align the behavior of DNN_DIAGNOSTICS_RUN between onnx and tf importers * move skipModelInput * print all missing layers * address TF issue
30 lines
801 B
C++
30 lines
801 B
C++
// 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
|
|
//! @{
|
|
|
|
typedef std::map<std::string, std::vector<LayerFactory::Constructor> > LayerFactory_Impl;
|
|
|
|
//! Register layer types of DNN model.
|
|
//!
|
|
//! @note In order to thread-safely access the factory, see getLayerFactoryMutex() function.
|
|
LayerFactory_Impl& getLayerFactoryImpl();
|
|
|
|
//! Get the mutex guarding @ref LayerFactory_Impl, see getLayerFactoryImpl() function.
|
|
Mutex& getLayerFactoryMutex();
|
|
|
|
//! @}
|
|
CV__DNN_INLINE_NS_END
|
|
}
|
|
}
|
|
#endif
|