• Varun Agrawal's avatar
    nms_cuda signature update (#945) · 249cfbf5
    Varun Agrawal authored
    Updated nms_cuda signature to accept detections and scores as separate tensors.
    This also required updating the indexing in the NMS CUDA kernel.
    
    Also made the iou_threshold parameter name consistent across implementations.
    249cfbf5
nms_cpu.cpp 2.46 KB