Added .webp .bmp

This commit is contained in:
alg-wiki
2022-10-11 06:33:08 +09:00
parent f0ab972f85
commit 907a88b2d0
3 changed files with 3 additions and 3 deletions

View File

@@ -22,7 +22,7 @@ class PersonalizedBase(Dataset):
self.width = width
self.height = height
self.flip = transforms.RandomHorizontalFlip(p=flip_p)
self.extns = [".jpg",".jpeg",".png"]
self.extns = [".jpg",".jpeg",".png",".webp",".bmp"]
self.dataset = []