potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 15:53:03 +04:00
Code Packages Releases Wiki Activity

add celu, hardsigmoid, selu, thresholdedrelu layers

Browse Source
This commit is contained in:
Smirnov Egor
2021-12-03 23:17:07 +03:00
parent defd8a527b
commit 71a22e45b0
10 changed files with 526 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/include/opencv2/dnn/all_layers.hpp
+34
View File
@@ -738,6 +738,40 @@ CV__DNN_INLINE_NS_BEGIN
static Ptr<TanLayer> create(const LayerParams &params);
};
class CV_EXPORTS CeluLayer : public ActivationLayer
{
public:
float alpha;
static Ptr<CeluLayer> create(const LayerParams &params);
};
class CV_EXPORTS HardSigmoidLayer : public ActivationLayer
{
public:
float alpha;
float beta;
static Ptr<HardSigmoidLayer> create(const LayerParams &params);
};
class CV_EXPORTS SeluLayer : public ActivationLayer
{
public:
float alpha;
float gamma;
static Ptr<SeluLayer> create(const LayerParams &params);
};
class CV_EXPORTS ThresholdedReluLayer : public ActivationLayer
{
public:
float alpha;
static Ptr<ThresholdedReluLayer> create(const LayerParams &params);
};
class CV_EXPORTS ActivationLayerInt8 : public ActivationLayer
{
public: