C++ extension implementation of RoIAlign & RolPool (both GPU and CPU) for PyTorch
Primary LanguageC++