assert self.groups[0].item() == 0 and self.groups[-1].item() == len(self.groups) - 1

[cruigo93]

❓ Questions and Help

General questions about detectron2.

I have binary masks for objects and I just make little changes to demo code to train Mask-RCNN. Visualization is fine but I cannot train model

[ppwwyyxx]

It looks like all your images have similar aspect ratio (all width > height, or the other way around) and in this case you cannot use cfg.DATALOADER.ASPECT_RATIO_GROUPING (default to True). Setting it to False should work. We'll also address it in the code so it does not throw an error.