I’d like to use a pure image and an arbitrary mask to mask the original image so that i can get a masked image.
So it means that I can combine A and B and the true len of dataset is len(A)*len(B)
but I reckon ur code example a good one,the len(dataset)=len(A), and mask is random sampled by the randint func.