From 38ad5a4644b1c38c7667823ec248cd8deb29613a Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 7 Apr 2025 12:46:41 -0600 Subject: [PATCH] Fixed issue with video dataset sizing --- toolkit/data_transfer_object/data_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/data_transfer_object/data_loader.py b/toolkit/data_transfer_object/data_loader.py index ad9a8fdd..26c1054c 100644 --- a/toolkit/data_transfer_object/data_loader.py +++ b/toolkit/data_transfer_object/data_loader.py @@ -62,7 +62,7 @@ class FileItemDTO( use_db_entry = False if file_key in size_database: db_entry = size_database[file_key] - if db_entry is not None and db_entry[2] == file_signature: + if db_entry is not None and len(db_entry) >= 3 and db_entry[2] == file_signature: use_db_entry = True if use_db_entry: @@ -81,7 +81,7 @@ class FileItemDTO( # Release the video capture object immediately video.release() - size_database[file_key] = (width, height) + size_database[file_key] = (width, height, file_signature) else: # original method is significantly faster, but some images are read sideways. Not sure why. Do slow method for now. # process width and height