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

Merge 51ae3893a7 into 8609e5fae5

Browse Source
This commit is contained in:
Hjosmelvy 2023-12-27 07:06:50 +00:00 committed by GitHub
commit d9770dc91d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
mediapipe/__pycache__/__init__.cpython-310.pyc Normal file
View File

Binary file not shown.

BIN
mediapipe/python/__pycache__/__init__.cpython-310.pyc Normal file
View File

Binary file not shown.

15
mediapipe/python/solution_base.py
View File

@ -373,17 +373,14 @@ class SolutionBase:
# Create a NamedTuple object where the field names are mapping to the graph # Create a NamedTuple object where the field names are mapping to the graph
# output stream names. # output stream names.
solution_outputs = collections.namedtuple( solution_outputs = collections.namedtuple(
'SolutionOutputs', self._output_stream_type_info.keys()) 'SolutionOutputs', self._output_stream_type_info.keys())
results = solution_outputs(*([""] * len(solution_outputs._fields)))
for stream_name in self._output_stream_type_info.keys(): for stream_name in self._output_stream_type_info.keys():
if stream_name in self._graph_outputs: if stream_name in self._graph_outputs:
setattr( results = results._replace(**{stream_name: self._get_packet_content(
solution_outputs, stream_name, self._output_stream_type_info[stream_name], self._graph_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 return results
def close(self) -> None: def close(self) -> None:
"""Closes all the input sources and the graph.""" """Closes all the input sources and the graph."""