diff --git a/plugin/efficientNMSPlugin/efficientNMSInference.cu b/plugin/efficientNMSPlugin/efficientNMSInference.cu index 81d48fb81..3cf7e8ef4 100644 --- a/plugin/efficientNMSPlugin/efficientNMSInference.cu +++ b/plugin/efficientNMSPlugin/efficientNMSInference.cu @@ -314,16 +314,16 @@ __global__ void EfficientNMS(EfficientNMSParameters param, const int* topNumData for (int tile = 0; tile < numTiles; tile++) { - bool check_class = true; + bool ignoreClass = true; if (!param.classAgnostic) - check_class = threadClass[tile] == testClass; + ignoreClass = threadClass[tile] == testClass; // IOU if (boxIdx[tile] > i && // Make sure two different boxes are being tested, and that it's a higher index; boxIdx[tile] < numSelectedBoxes && // Make sure the box is within numSelectedBoxes; blockState == 1 && // Signal that allows IOU checks to be performed; threadState[tile] == 0 && // Make sure this box hasn't been either dropped or kept already; - check_class && // Compare only boxes of matching classes when classAgnostic is false; + ignoreClass && // Compare only boxes of matching classes when classAgnostic is false; lte_mp(threadScore[tile], testScore) && // Make sure the sorting order of scores is as expected; IOU(param, threadBox[tile], testBox) >= param.iouThreshold) // And... IOU overlap. {