From df1c10f00266184e402750deed4347ae3b51a0e1 Mon Sep 17 00:00:00 2001 From: SheffeyG <57262511+SheffeyG@users.noreply.github.com> Date: Thu, 10 Oct 2024 21:47:01 +0800 Subject: [PATCH] Fix empty paths while saving images with PIL bytes input (#16809) Co-authored-by: Laughing-q <1185102784@qq.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Glenn Jocher --- ultralytics/data/loaders.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/data/loaders.py b/ultralytics/data/loaders.py index e91f2082..34521692 100644 --- a/ultralytics/data/loaders.py +++ b/ultralytics/data/loaders.py @@ -409,7 +409,8 @@ class LoadPilAndNumpy: """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)