From 9794416a5d389fb3888f0953dac9e1e25e9cb9b2 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 10 Apr 2025 08:16:05 -0600 Subject: [PATCH] Fixed bug when loading video datasets --- toolkit/data_transfer_object/data_loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/toolkit/data_transfer_object/data_loader.py b/toolkit/data_transfer_object/data_loader.py index 26c1054c..ba235e91 100644 --- a/toolkit/data_transfer_object/data_loader.py +++ b/toolkit/data_transfer_object/data_loader.py @@ -78,6 +78,7 @@ class FileItemDTO( # Get width and height width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH)) height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT)) + w, h = width, height # Release the video capture object immediately video.release()