mirror of
https://github.com/opencv/opencv.git
synced 2026-07-30 15:53:03 +04:00
Merge pull request #28522 from abhishek-gola:batchnorm_layer_add
Added Batch Normalization layer to new DNN engine
This commit is contained in:
@@ -1161,6 +1161,15 @@ CV__DNN_INLINE_NS_BEGIN
|
||||
static Ptr<BatchNormLayer> create(const LayerParams ¶ms);
|
||||
};
|
||||
|
||||
class CV_EXPORTS BatchNorm2Layer : public Layer
|
||||
{
|
||||
public:
|
||||
float epsilon;
|
||||
bool useGlobalStats, hasWeights, hasBias;
|
||||
|
||||
static Ptr<BatchNorm2Layer> create(const LayerParams& params);
|
||||
};
|
||||
|
||||
class CV_EXPORTS BatchNormLayerInt8 : public BatchNormLayer
|
||||
{
|
||||
public:
|
||||
|
||||
Reference in New Issue
Block a user