From 55e604e6f751bce08beded401865e1b84523f6d4 Mon Sep 17 00:00:00 2001
From: BaofengZan <zanbaofeng1123@gmail.com>
Date: Fri, 9 Oct 2020 17:35:13 +0800
Subject: [PATCH] Fix bug : when the width and height are different, the result
 is incorrect (#235)

---
 yolov5/yololayer.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yolov5/yololayer.cu b/yolov5/yololayer.cu
index b3ed7c3..443d1ce 100644
--- a/yolov5/yololayer.cu
+++ b/yolov5/yololayer.cu
@@ -230,7 +230,7 @@ namespace nvinfer1
 
             //printf("Net: %d  %d \n", mYoloV5NetWidth, mYoloV5NetHeight);
             CalDetection << < (yolo.width*yolo.height*batchSize + mThreadCount - 1) / mThreadCount, mThreadCount >> >
-                (inputs[i], output, numElem, mYoloV5NetWidth, mYoloV5NetWidth, mMaxOutObject, yolo.width, yolo.height, (float *)mAnchor[i], mClassCount, outputElem);
+                (inputs[i], output, numElem, mYoloV5NetWidth, mYoloV5NetHeight, mMaxOutObject, yolo.width, yolo.height, (float *)mAnchor[i], mClassCount, outputElem);
         }
     }