Thanks @spanev

So what are the arguments image_mean, image_std used for?
Why should we pass boxes if we already pass segmentation masks? The boxes can be calculated from the maps. Am I missing something?