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

fix add_packet_to_input_stream deadlock of python

Browse Source
This commit is contained in:
jiangzhiwen01 2021-11-06 17:38:15 +08:00
parent f4e7f6cc48
commit e4275063fe
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/python/pybind/calculator_graph.cc
View File

@ -168,6 +168,7 @@ void CalculatorGraphSubmodule(pybind11::module* module) {
RaisePyErrorIfNotOk( RaisePyErrorIfNotOk(
self->AddPacketToInputStream(stream, packet.At(packet_timestamp))); self->AddPacketToInputStream(stream, packet.At(packet_timestamp)));
}, },
py::call_guard<py::gil_scoped_release>(),
R"doc(Add a packet to a graph input stream. R"doc(Add a packet to a graph input stream.
If the graph input stream add mode is ADD_IF_NOT_FULL, the packet will not be If the graph input stream add mode is ADD_IF_NOT_FULL, the packet will not be
@ -347,6 +348,7 @@ void CalculatorGraphSubmodule(pybind11::module* module) {
calculator_graph.def( calculator_graph.def(
"wait_for_observed_output", "wait_for_observed_output",
[](CalculatorGraph* self) { [](CalculatorGraph* self) {
py::gil_scoped_release gil_release;
RaisePyErrorIfNotOk(self->WaitForObservedOutput()); RaisePyErrorIfNotOk(self->WaitForObservedOutput());
}, },
R"doc(Wait until a packet is emitted on one of the observed output streams. R"doc(Wait until a packet is emitted on one of the observed output streams.