No problem. Inputs in a single batch item are:
image - tensor[3,520,704]
bbox - tensor[-1, 4]
mask - tensor[1, 520,704]
Sartorius Kaggle competition is the dataset when this bug occurred, if you need some sample data, I’ll provide it as well.