Closed den-107 closed 1 year ago
Triple the channels of your images or change the line 101 in Data.py to open the images as RGB:
img = Image.open(join(self.img_dir, image_fn)).convert('RGB')
Tha you for answering this @MohamadHawshar! Yes DINO pretrained only works with RGB images so you need to bump up the channels or make your on special DINO
Hello, anyone had luck training stego with greyscale images? In my case it seems I cant get the needed pretrained weights for the DinoFeaturizer class. Anyone has a url to load these weights?