yolo3 ok

src/Yolo.cpp

@@ -24,7 +24,7 @@ Yolo::Yolo(Network *net, int classes, int num) :
     output_dim.w = input_dim.w;
     output_dim.l = input_dim.l;
 
-    checkCuda( cudaMalloc(&dstData, input_dim.tot()*sizeof(dnnType)) );
+    checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
 }
 
 Yolo::~Yolo() {