Add YOLOX support

2023-01-30 23:59:51 -03:00
parent f9c7a4dfca
commit 825d6bfda8
11 changed files with 746 additions and 24 deletions
--- a/nvdsinfer_custom_impl_Yolo/yoloForward_v8.cu
+++ b/nvdsinfer_custom_impl_Yolo/yoloForward_v8.cu
@@ -18,7 +18,7 @@ __global__ void gpuYoloLayer_v8(const float* input, int* num_detections, float*
  int maxIndex = -1;

  for (uint i = 0; i < numOutputClasses; ++i) {
-    float prob = input[x_id * (4 + numOutputClasses) + i + 4];
+    float prob = input[x_id * (4 + numOutputClasses) + 4 + i];
    if (prob > maxProb) {
      maxProb = prob;
      maxIndex = i;