Fix empty paths while saving images with PIL bytes input (ultralytics…

…#16809) Co-authored-by: Laughing-q <[email protected]> Co-authored-by: Laughing <[email protected]> Co-authored-by: Glenn Jocher <[email protected]>
Ashkanbhz · Oct 10, 2024 · df1c10f · df1c10f
1 parent 3f341d5
commit df1c10f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ultralytics/data/loaders.py b/ultralytics/data/loaders.py
@@ -409,7 +409,8 @@ def __init__(self, im0):
         """Initialize PIL and Numpy Dataloader."""
         if not isinstance(im0, list):
             im0 = [im0]
-        self.paths = [getattr(im, "filename", f"image{i}.jpg") for i, im in enumerate(im0)]
+        # use `image{i}.jpg` when Image.filename returns an empty path.
+        self.paths = [getattr(im, "filename", "") or f"image{i}.jpg" for i, im in enumerate(im0)]
         self.im0 = [self._single_check(im) for im in im0]
         self.mode = "image"
         self.bs = len(self.im0)