ITS: Check if there are any extracted frames.
During Video ITS, we extract key frames from the recorded video.
Add an assertion to check if there were any frames extracted.
In cases where there were no extracted frames, video source should
be checked. We noticed this for 480P video recording.
Bug: 229773134
Test: Tested locally on pixel device.
Change-Id: Ie3ad482cba2f3b201142cb00658f3cf9f8ab3711
diff --git a/apps/CameraITS/utils/video_processing_utils.py b/apps/CameraITS/utils/video_processing_utils.py
index 894729c..187917a 100644
--- a/apps/CameraITS/utils/video_processing_utils.py
+++ b/apps/CameraITS/utils/video_processing_utils.py
@@ -74,6 +74,10 @@
for file in arr:
if '.png' in file and not os.path.isdir(file) and ffmpeg_image_name in file:
key_frame_files.append(file)
+
+ if not len(key_frame_files):
+ raise AssertionError('No key frames extracted. Check source video.')
+
return key_frame_files
@@ -124,4 +128,7 @@
file_list = sorted(
[_ for _ in os.listdir(log_path) if (_.endswith(img_format)
and ffmpeg_image_name in _)])
+ if not len(file_list):
+ raise AssertionError('No frames extracted. Check source video.')
+
return file_list
