cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

results correctly added to NamedTuple

Browse Source
This commit is contained in:
hjosmelvy 2023-01-23 18:46:48 -05:00
parent 873d7181bf
commit 4e79b53ffd
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mediapipe/python/solution_base.py
View File

@ -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)
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 solution_outputs
return results
def close(self) -> None:
"""Closes all the input sources and the graph."""