RoI Align Module for Mask R-CNN

[GoktugGuvercin]

Hello Flax Community,

I aim to implement Mask R-CNN and use it for my document segmentation task by using Flax. As we know, one of core modules in this segmentation model is Roi-Align, but I could not find its implementation in Flax.

• What do you recommend at this point ?
• Does Flax have Roi-Align implementation in its ecosystem ?

Thanks in advance.