I have a dataset that consists of two classes in different folders. Annotation provided in the form of binary mask images. Can you recommend information that will help me tr