talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: c27bb068d24890a4e3b7d2d536267e08d7e4abbd
This commit is contained in:
MediaPipe Team
2020-08-31 16:03:43 -07:00
committed by jqtang
parent c0124fb83c
commit 1db91b550a
5 changed files with 24 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/calculators/tensorflow/tensorflow_session_from_frozen_graph_calculator.cc
+1 -1
View File
@@ -51,7 +51,7 @@ namespace {
void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) {
for (auto& node : *graph_def->mutable_node()) {
if (node.device().empty()) {
node.set_device(device_id);
node.set_device(std::string(device_id));
}
}
}
mediapipe/calculators/tensorflow/tensorflow_session_from_frozen_graph_generator.cc
+1 -1
View File
@@ -47,7 +47,7 @@ namespace {
void SetPreferredDevice(tf::GraphDef* graph_def, absl::string_view device_id) {
for (auto& node : *graph_def->mutable_node()) {
if (node.device().empty()) {
node.set_device(device_id);
node.set_device(std::string(device_id));
}
}
}
mediapipe/calculators/tensorflow/unpack_media_sequence_calculator.proto
+7
View File
@@ -56,4 +56,11 @@ message UnpackMediaSequenceCalculatorOptions {
// the clip start and end times and outputs these for the
// AudioDecoderCalculator to consume.
optional AudioDecoderOptions base_audio_decoder_options = 9;
optional string keypoint_names = 10 [
default =
"NOSE,LEFT_EAR,RIGHT_EAR,LEFT_SHOULDER,RIGHT_SHOULDER,LEFT_FORE_PAW,RIGHT_FORE_PAW,LEFT_HIP,RIGHT_HIP,LEFT_HIND_PAW,RIGHT_HIND_PAW,ROOT_TAIL"
];
// When the keypoint doesn't exists, output this default value.
optional float default_keypoint_location = 11 [default = -1.0];
}