Fix screenshot inference (#4033)

2 years ago · 99cb549efe
parent e346b59658
commit 99cb549efe
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/data/loaders.py
+++ b/ultralytics/data/loaders.py
@ -156,7 +156,7 @@ class LoadScreenshots:
        s = f'screen {self.screen} (LTWH): {self.left},{self.top},{self.width},{self.height}: '

        self.frame += 1
-        return str(self.screen), im0, None, s  # screen, img, original img, im0s, s
+        return [str(self.screen)], [im0], None, s  # screen, img, vid_cap, string


 class LoadImages: