How to implement Non Max Suppression (NMS) in PyTorch - View it on GitHub