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
88 lines
2.3 KiB
C++
88 lines
2.3 KiB
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_COMMON_HPP__
|
|
#define __OPENCV_DNN_COMMON_HPP__
|
|
|
|
#include <unordered_set>
|
|
#include <unordered_map>
|
|
|
|
#include <opencv2/dnn.hpp>
|
|
|
|
namespace cv { namespace dnn {
|
|
CV__DNN_INLINE_NS_BEGIN
|
|
#define IS_DNN_OPENCL_TARGET(id) (id == DNN_TARGET_OPENCL || id == DNN_TARGET_OPENCL_FP16)
|
|
Mutex& getInitializationMutex();
|
|
void initializeLayerFactory();
|
|
|
|
extern bool DNN_DIAGNOSTICS_RUN;
|
|
extern bool DNN_SKIP_REAL_IMPORT;
|
|
|
|
namespace detail {
|
|
#define CALL_MEMBER_FN(object, ptrToMemFn) ((object).*(ptrToMemFn))
|
|
|
|
class NotImplemented : public Layer
|
|
{
|
|
public:
|
|
static Ptr<Layer> create(const LayerParams ¶ms);
|
|
|
|
static void Register();
|
|
static void unRegister();
|
|
};
|
|
|
|
template <typename Importer, typename ... Args>
|
|
Net readNet(Args&& ... args)
|
|
{
|
|
Net net;
|
|
Importer importer(net, std::forward<Args>(args)...);
|
|
return net;
|
|
}
|
|
|
|
template <typename Importer, typename ... Args>
|
|
Net readNetDiagnostic(Args&& ... args)
|
|
{
|
|
Net maybeDebugNet = readNet<Importer>(std::forward<Args>(args)...);
|
|
if (DNN_DIAGNOSTICS_RUN && !DNN_SKIP_REAL_IMPORT)
|
|
{
|
|
// if we just imported the net in diagnostic mode, disable it and import again
|
|
enableModelDiagnostics(false);
|
|
Net releaseNet = readNet<Importer>(std::forward<Args>(args)...);
|
|
enableModelDiagnostics(true);
|
|
return releaseNet;
|
|
}
|
|
return maybeDebugNet;
|
|
}
|
|
|
|
class LayerHandler
|
|
{
|
|
public:
|
|
void addMissing(const std::string& name, const std::string& type);
|
|
bool contains(const std::string& type) const;
|
|
void printMissing();
|
|
|
|
protected:
|
|
LayerParams getNotImplementedParams(const std::string& name, const std::string& op);
|
|
|
|
private:
|
|
std::unordered_map<std::string, std::unordered_set<std::string>> layers;
|
|
};
|
|
|
|
struct NetImplBase
|
|
{
|
|
const int networkId; // network global identifier
|
|
int networkDumpCounter; // dump counter
|
|
int dumpLevel; // level of information dumps (initialized through OPENCV_DNN_NETWORK_DUMP parameter)
|
|
|
|
NetImplBase();
|
|
|
|
std::string getDumpFileNameBase();
|
|
};
|
|
|
|
} // namespace detail
|
|
|
|
CV__DNN_INLINE_NS_END
|
|
}} // namespace
|
|
|
|
#endif // __OPENCV_DNN_COMMON_HPP__
|