Add PP-YOLOE support

2022-07-24 18:00:47 -03:00
parent d09879d557
commit a3782ed65e
51 changed files with 1812 additions and 600 deletions
--- a/nvdsinfer_custom_impl_Yolo/yoloForward.cu
+++ b/nvdsinfer_custom_impl_Yolo/yoloForward.cu
@@ -28,7 +28,7 @@ __global__ void gpuYoloLayer(
    if (objectness < scoreThreshold)
        return;

-    int count = (int)atomicAdd(&countData[0], 1);
+    int count = (int)atomicAdd(countData, 1);

    const float alpha = scaleXY;
    const float beta = -0.5 * (scaleXY - 1);