New YOLOv5 conversion and support (>= v2.0)

nvdsinfer_custom_impl_Yolo/layers/batchnorm_layer.h

@@ -0,0 +1,27 @@
+/*
+ * Created by Marcos Luciano
+ * https://www.github.com/marcoslucianops
+ */
+
+#ifndef __BATCHNORM_LAYER_H__
+#define __BATCHNORM_LAYER_H__
+
+#include <map>
+#include <vector>
+
+#include "NvInfer.h"
+
+#include "activation_layer.h"
+
+nvinfer1::ILayer* batchnormLayer(
+    int layerIdx,
+    std::map<std::string, std::string>& block,
+    std::vector<float>& weights,
+    std::vector<nvinfer1::Weights>& trtWeights,
+    int& weightPtr,
+    std::string weightsType,
+    float eps,
+    nvinfer1::ITensor* input,
+    nvinfer1::INetworkDefinition* network);
+
+#endif