exception.py

class ImageFolderEX(datasets.ImageFolder):
    def __getitem__(self, index):
        while True:
            path, label = self.imgs[index]
            try:
                img = self.loader(os.path.join(self.root, path))
                if self.transform is not None:
                    img = self.transform(img)
            except:
                index = np.random.randint(self.__len__())
                continue
            break
        return [img, label]