yolo layers

src/Reorg.cpp

@@ -0,0 +1,35 @@
+#include <iostream>
+
+#include "Layer.h"
+#include "kernels.h"
+
+namespace tkDNN {
+
+Reorg::Reorg(Network *net, dataDim_t input_dim, int stride) : 
+    Layer(net, input_dim) {
+
+    this->stride = stride;
+    
+    output_dim.n = input_dim.n;
+    output_dim.c = input_dim.c*stride*stride;
+    output_dim.h = input_dim.h/stride;
+    output_dim.w = input_dim.w/stride;
+    output_dim.l = input_dim.l;
+    
+    checkCuda( cudaMalloc(&dstData, input_dim.tot()*sizeof(value_type)) );
+}
+
+Reorg::~Reorg() {
+
+    checkCuda( cudaFree(dstData) );
+}
+
+value_type* Reorg::infer(dataDim_t &dim, value_type* srcData) {
+
+    reorgForward(srcData, dstData, dim, stride);
+
+    dim = output_dim;
+    return dstData;
+}
+
+}