From 4e79b53ffdd1ba69d49dd7c23bae633017450fc5 Mon Sep 17 00:00:00 2001 From: hjosmelvy Date: Mon, 23 Jan 2023 18:46:48 -0500 Subject: [PATCH] results correctly added to NamedTuple --- mediapipe/python/solution_base.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/mediapipe/python/solution_base.py b/mediapipe/python/solution_base.py index 020d16371..791340ab3 100644 --- a/mediapipe/python/solution_base.py +++ b/mediapipe/python/solution_base.py @@ -366,17 +366,15 @@ class SolutionBase: # Create a NamedTuple object where the field names are mapping to the graph # output stream names. solution_outputs = collections.namedtuple( - 'SolutionOutputs', self._output_stream_type_info.keys()) + 'SolutionOutputs', self._output_stream_type_info.keys()) + empty_list = [[] for _ in range(len(solution_outputs._fields))] + results = solution_outputs(*(empty_list)) for stream_name in self._output_stream_type_info.keys(): - if stream_name in self._graph_outputs: - setattr( - solution_outputs, stream_name, - self._get_packet_content(self._output_stream_type_info[stream_name], - self._graph_outputs[stream_name])) - else: - setattr(solution_outputs, stream_name, None) - - return solution_outputs + if stream_name in self._graph_outputs: + results = results._replace(**{stream_name: self._get_packet_content( + self._output_stream_type_info[stream_name], self._graph_outputs[stream_name])}) + + return results def close(self) -> None: """Closes all the input sources and the graph."""