torchrgb

mwalmsley · Jan 5, 2024 · 862ee37 · 862ee37
1 parent 8d15167
commit 862ee37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zoobot/pytorch/datasets/webdatamodule.py b/zoobot/pytorch/datasets/webdatamodule.py
@@ -118,7 +118,7 @@ def make_loader(self, urls, mode="train"):
             dataset = dataset.shuffle(shuffle)
 
         # dataset = dataset.decode("rgb")  # np.array, for albumentations
-        dataset = dataset.decode("pilrgb")  # PIL Image, for torchvision
+        dataset = dataset.decode("torchrgb")  # tensor, for torchvision
 
         if mode == 'predict':
             if self.label_cols != ['id_str']: