Pytorch Implementation of RetinaNet with CUDA accelerate nms operation. - View it on GitHub