Merge pull request #2288 from activeloopai/change_pil_byte_calc

[AL-2215] Change byte calculation for PIL image in shuffle buffer

deeplake/integrations/pytorch/shuffle_buffer.py

@@ -128,9 +128,16 @@ def _sample_size(self, sample):
         elif isinstance(sample, np.ndarray):
             return sample.nbytes
         elif isinstance(sample, Image.Image):
-            img = BytesIO()
-            sample.save(img, sample.format)
-            return len(img.getvalue())
+            size = sample.size
+            num_pixels = size[0] * size[1]
+            if sample.mode == "RGB":
+                num_pixels = num_pixels * 3
+            elif sample.mode == "RGBA":
+                num_pixels = num_pixels * 4
+            elif sample.mode == "L":
+                num_pixels = num_pixels * 1
+            num_bytes = num_pixels * 1  # change according to dtype of tensor later
+            return num_bytes
         raise ValueError(
             f"Expected input of type bytes, dict, Sequence, torch.Tensor, np.ndarray or PIL image, got: {type(sample)}"
         )