Import dataset without labels

Hi everyone!
I’m trying to import the AITEX Dataset that hasn’t got labels because in the folder there are just images. The structure of the folder content is:

  • AITEX
    – Defect_images → it contains images
    – Mask_images → it contains images
    – NODefect_images → it contains subdirectories with images

How can i write class CustomTextDataset(Dataset)?