Update setattr() default for Hub PIL images (#3923)

Fix inference from PIL source.
ultralytics · Jul 7, 2021 · 588094e · 588094e
1 parent 411842e
commit 588094e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/common.py b/models/common.py
@@ -255,7 +255,7 @@ def forward(self, imgs, size=640, augment=False, profile=False):
                 im, f = Image.open(requests.get(im, stream=True).raw if im.startswith('http') else im), im
                 im = np.asarray(exif_transpose(im))
             elif isinstance(im, Image.Image):  # PIL Image
-                im, f = np.asarray(exif_transpose(im)), getattr(im, 'filename') or f
+                im, f = np.asarray(exif_transpose(im)), getattr(im, 'filename', f) or f
             files.append(Path(f).with_suffix('.jpg').name)
             if im.shape[0] < 5:  # image in CHW
                 im = im.transpose((1, 2, 0))  # reverse dataloader .transpose(2, 0, 1)