Support yolov4

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>

include/tkDNN/pluginsRT/ActivationMishRT.h

@@ -0,0 +1,60 @@
+#include<cassert>
+#include "../kernels.h"
+
+class ActivationMishRT : public IPlugin {
+
+public:
+	ActivationMishRT() {
+
+
+	}
+
+	~ActivationMishRT(){
+
+	}
+
+	int getNbOutputs() const override {
+		return 1;
+	}
+
+	Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override {
+		return inputs[0];
+	}
+
+	void configure(const Dims* inputDims, int nbInputs, const Dims* outputDims, int nbOutputs, int maxBatchSize) override {
+		size = 1;
+		for(int i=0; i<outputDims[0].nbDims; i++)
+			size *= outputDims[0].d[i];
+	}
+
+	int initialize() override {
+
+		return 0;
+	}
+
+	virtual void terminate() override {
+	}
+
+	virtual size_t getWorkspaceSize(int maxBatchSize) const override {
+		return 0;
+	}
+
+	virtual int enqueue(int batchSize, const void*const * inputs, void** outputs, void* workspace, cudaStream_t stream) override {
+
+		activationMishForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]), 
+											reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, stream);
+		return 0;
+	}
+
+
+	virtual size_t getSerializationSize() override {
+		return 1*sizeof(int);
+	}
+
+	virtual void serialize(void* buffer) override {
+		char *buf = reinterpret_cast<char*>(buffer);
+		tk::dnn::writeBUF(buf, size);
+	}
+
+	int size;
+};

include/tkDNN/pluginsRT/YoloRT.h

@@ -8,11 +8,12 @@ class YoloRT : public IPlugin {
 
 
 public:
-	YoloRT(int classes, int num, tk::dnn::Yolo *yolo = nullptr, int n_masks=3) {
+	YoloRT(int classes, int num, tk::dnn::Yolo *yolo = nullptr, int n_masks=3, float scale_xy=1) {
 
 		this->classes = classes;
 		this->num = num;
 		this->n_masks = n_masks;
+		this->scaleXY = scale_xy;
 
         mask = new dnnType[n_masks];
         bias = new dnnType[num*n_masks*2];
@@ -64,6 +65,8 @@ public:
 			for(int n = 0; n < n_masks; ++n){
 				int index = entry_index(b, n*w*h, 0);
 				activationLOGISTICForward(srcData + index, dstData + index, 2*w*h, stream);
+
+				if (this->scaleXY != 1) scalAdd(dstData + index, 2 * w*h, this->scaleXY, -0.5*(this->scaleXY - 1), 1);
 				
 				index = entry_index(b, n*w*h, 4);
 				activationLOGISTICForward(srcData + index, dstData + index, (1+classes)*w*h, stream);
@@ -76,7 +79,7 @@ public:
 
 
 	virtual size_t getSerializationSize() override {
-		return 6*sizeof(int) + n_masks*sizeof(dnnType) + num*n_masks*2*sizeof(dnnType) + YOLORT_CLASSNAME_W*classes*sizeof(char);
+		return 6*sizeof(int) + sizeof(float)+ n_masks*sizeof(dnnType) + num*n_masks*2*sizeof(dnnType) + YOLORT_CLASSNAME_W*classes*sizeof(char);
 	}
 
 	virtual void serialize(void* buffer) override {
@@ -87,6 +90,7 @@ public:
 		tk::dnn::writeBUF(buf, c);
 		tk::dnn::writeBUF(buf, h);
 		tk::dnn::writeBUF(buf, w);
+		tk::dnn::writeBUF(buf, scaleXY);
         for(int i=0; i<n_masks; i++)
     		tk::dnn::writeBUF(buf, mask[i]);
         for(int i=0; i<n_masks*2*num; i++)
@@ -104,6 +108,7 @@ public:
 
 	int c, h, w;
     int classes, num, n_masks;
+	float scaleXY;
 	std::vector<std::string> classesNames;
 
     dnnType *mask;