From dc36015acbb30943259069c5b493254f51bb2a57 Mon Sep 17 00:00:00 2001 From: Joseph VanPelt Date: Mon, 4 Nov 2024 09:45:02 -0500 Subject: [PATCH] fix for issue where the window has no frames yet --- .../activity_classification/activity_classifier_tcn.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ros/angel_system_nodes/angel_system_nodes/activity_classification/activity_classifier_tcn.py b/ros/angel_system_nodes/angel_system_nodes/activity_classification/activity_classifier_tcn.py index 2aac71f75..6c0287453 100644 --- a/ros/angel_system_nodes/angel_system_nodes/activity_classification/activity_classifier_tcn.py +++ b/ros/angel_system_nodes/angel_system_nodes/activity_classification/activity_classifier_tcn.py @@ -592,8 +592,10 @@ def rt_loop(self): have_leading_object=self._window_lead_with_objects, ) - window_end_frame = window.frames[-1][0] - image_gid = self._collect_image(window_end_frame) + image_gid = None + if len(window.frames) > 0 and len(window.frames[-1]) > 0: + window_end_frame = window.frames[-1][0] + image_gid = self._collect_image(window_end_frame) # log.info(f"buffer contents: {window.obj_dets}")