Skip to content

Commit

Permalink
change variable name
Browse files Browse the repository at this point in the history
Signed-off-by: yili123123 <yiliyili123@outlook.com>
  • Loading branch information
yili123123 committed Feb 12, 2023
1 parent 6d52bd9 commit 15338f4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions plugin/efficientNMSPlugin/efficientNMSInference.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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<T>(param, threadBox[tile], testBox) >= param.iouThreshold) // And... IOU overlap.
{
Expand Down

0 comments on commit 15338f4

Please sign in to comment.