About custom dataset.

[willpat1213]

There is a custom dataset, each picture contains multiple objects, how to modify the dataloader or model to train such a dataset?

[ymq2017]

Hi, for a multi-object dataset, you could simply modify the dataloader to sample one mask each time randomly.
Or you could load all masks and use them as prompts. Then we can change the prompt input to multi-mask, multi-box, multi-points. And the multi-mask output can be used to calculate loss with input multi GT mask.
The latter requires a larger code change and would work better.

[willpat1213]

Thanks for your reply!
Another question is how do I design the loss in this case?