Skip to main content

roi_align

Function roi_align 

Source
pub fn roi_align(
    features: &Tensor,
    rois: &[(f32, f32, f32, f32)],
    output_size: (usize, usize),
    sampling_ratio: usize,
) -> Result<Tensor, DetectError>
Expand description

RoI Align: bilinear-interpolation version of RoI pooling (no quantisation).

  • features[H, W, C] single-image feature map.
  • rois – slice of (x1, y1, x2, y2) in feature-map coordinates.
  • output_size(out_h, out_w).
  • sampling_ratio – number of sampling points per bin dimension (0 means adaptive: ceil(bin_size)).

Returns a tensor of shape [num_rois, out_h, out_w, C].