Updating mask_roi_align for multiple masks

[EEvangelisti]

Hello,
I am trying to adapt your very nice implementation to my needs. Basically, I have multiple (2 to 5) shapes on each image that I would like to identify. Unfortunately, the current implementation of mask_roi_align is designed for a single shape/mask per image and raises ValueError: could not broadcast input array due to the following line:
maskrois=np.zeros((batchlen,proposal_count,mask_size[0],mask_size[1],1))
Would you be able to suggest a way to allow for a variable number of masks, i.e. a more general value for the last dimension? Many thanks for your help.