Skip to main content
Log in

Mojo module

roi_align

Structs

  • Weighted2DPoint: Utility class to wrap 2-d point coordinates and floating point weight for bilinear interpolation.

Functions

  • roi_align_nhwc: Compute ROIAlign a batch of rois of shape [M, 5] where the first dim is the batch index, followed by region box coordinates (y0, x0) (y1, x1). For inputs of NHWC format. The output shape is [M, output_height, output_width, C].