text: from torch.nn.modules.module import Module
  from extensions.layers.roi.roi_pooling.functions.roi_pool import RoIPoolFunction
  
  
  class _RoIPooling(Module):
      def __init__(self, pooled_height, pooled_width, spatial_scale):
          super(_RoIPooling, self).__init__()
  
          self.pooled_width = int(pooled_width)
          self.pooled_height = int(pooled_height)
          self.spatial_scale = float(spatial_scale)
  
      def forward(self, features, rois):
          return RoIPoolFunction(self.pooled_height, self.pooled_width, self.spatial_scale)(features, rois)
  