RoIAlign.pytorch

github.com ↗

This is a PyTorch version of RoIAlign. This implementation is based on cropandresize and supports both forward and backward on CPU and GPU.