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

Pose detector uses advanced_gpu_api for gpu inference to resolve unsupported gpu op issue.

Browse Source 
PiperOrigin-RevId: 528879218
This commit is contained in:
Jiuqiang Tang 2023-05-02 13:49:24 -07:00 committed by Copybara-Service
parent 9ce16fddeb
commit 4d9812af43
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_graph.cc
View File

@ -108,9 +108,18 @@ absl::Status SetSubTaskBaseOptions(const ModelAssetBundleResources& resources,
->mutable_model_asset(),
is_copy);
}
pose_detector_graph_options->mutable_base_options()
->mutable_acceleration()
->CopyFrom(options->base_options().acceleration());
if (options->base_options().acceleration().has_gpu()) {
core::proto::Acceleration gpu_accel;
gpu_accel.mutable_gpu()->set_use_advanced_gpu_api(true);
pose_detector_graph_options->mutable_base_options()
->mutable_acceleration()
->CopyFrom(gpu_accel);
} else {
pose_detector_graph_options->mutable_base_options()
->mutable_acceleration()
->CopyFrom(options->base_options().acceleration());
}
pose_detector_graph_options->mutable_base_options()->set_use_stream_mode(
options->base_options().use_stream_mode());
auto* pose_landmarks_detector_graph_options =