diff --git a/modules/dnn/include/opencv2/dnn/all_layers.hpp b/modules/dnn/include/opencv2/dnn/all_layers.hpp index d9d1833780..24e792cb11 100644 --- a/modules/dnn/include/opencv2/dnn/all_layers.hpp +++ b/modules/dnn/include/opencv2/dnn/all_layers.hpp @@ -1204,6 +1204,12 @@ CV__DNN_INLINE_NS_BEGIN static Ptr create(const LayerParams& params); }; + class CV_EXPORTS RandomNormalLikeLayer : public Layer + { + public: + static Ptr create(const LayerParams& params); + }; + //! @} //! @} CV__DNN_INLINE_NS_END diff --git a/modules/dnn/src/init.cpp b/modules/dnn/src/init.cpp index 61db2c1ba6..eea68d6af1 100644 --- a/modules/dnn/src/init.cpp +++ b/modules/dnn/src/init.cpp @@ -200,6 +200,7 @@ void initializeLayerFactory() CV_DNN_REGISTER_LAYER_CLASS(ScatterND, ScatterNDLayer); CV_DNN_REGISTER_LAYER_CLASS(Tile, TileLayer); CV_DNN_REGISTER_LAYER_CLASS(TopK, TopKLayer); + CV_DNN_REGISTER_LAYER_CLASS(RandomNormalLike, RandomNormalLikeLayer); CV_DNN_REGISTER_LAYER_CLASS(Quantize, QuantizeLayer); CV_DNN_REGISTER_LAYER_CLASS(Dequantize, DequantizeLayer); diff --git a/modules/dnn/src/layers/randomnormallike_layer.cpp b/modules/dnn/src/layers/randomnormallike_layer.cpp index 30713390c9..4bd6ca5728 100644 --- a/modules/dnn/src/layers/randomnormallike_layer.cpp +++ b/modules/dnn/src/layers/randomnormallike_layer.cpp @@ -6,12 +6,11 @@ #include "../precomp.hpp" #include "layers_common.hpp" -#include #include namespace cv { namespace dnn { -class RandomNormalLikeLayerImpl CV_FINAL : public Layer +class RandomNormalLikeLayerImpl CV_FINAL : public RandomNormalLikeLayer { public: RandomNormalLikeLayerImpl(const LayerParams& params) @@ -117,6 +116,9 @@ private: int depth; }; -CV_DNN_REGISTER_LAYER_CLASS_STATIC(RandomNormalLike, RandomNormalLikeLayerImpl); +Ptr RandomNormalLikeLayer::create(const LayerParams& params) +{ + return makePtr(params); +} }} // namespace cv::dnn