From 99cb549efe36a2563d83aaf0bc5e2ad45601a73e Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 29 Jul 2023 22:53:28 +0200
Subject: [PATCH] Fix screenshot inference (#4033)

---
 ultralytics/data/loaders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ultralytics/data/loaders.py b/ultralytics/data/loaders.py
index 01acd8ee..a59e8c80 100644
--- 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: