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

Small fixes to TensorsToImageCalculator.

Browse Source 
PiperOrigin-RevId: 512967357
This commit is contained in:
Jiuqiang Tang 2023-02-28 10:08:37 -08:00 committed by Copybara-Service
parent 06dd003113
commit 646ab8505a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator.cc
View File

@ -92,7 +92,7 @@ class TensorsToImageCalculator : public Node {
MEDIAPIPE_NODE_CONTRACT(kInputTensors, kOutputImage); MEDIAPIPE_NODE_CONTRACT(kInputTensors, kOutputImage);
static absl::Status GetContract(CalculatorContract* cc); static absl::Status UpdateContract(CalculatorContract* cc);
absl::Status Open(CalculatorContext* cc); absl::Status Open(CalculatorContext* cc);
absl::Status Process(CalculatorContext* cc); absl::Status Process(CalculatorContext* cc);
absl::Status Close(CalculatorContext* cc); absl::Status Close(CalculatorContext* cc);
@ -124,7 +124,7 @@ class TensorsToImageCalculator : public Node {
}; };
MEDIAPIPE_REGISTER_NODE(::mediapipe::tasks::TensorsToImageCalculator); MEDIAPIPE_REGISTER_NODE(::mediapipe::tasks::TensorsToImageCalculator);
absl::Status TensorsToImageCalculator::GetContract(CalculatorContract* cc) { absl::Status TensorsToImageCalculator::UpdateContract(CalculatorContract* cc) {
#if !MEDIAPIPE_DISABLE_GPU #if !MEDIAPIPE_DISABLE_GPU
#if MEDIAPIPE_METAL_ENABLED #if MEDIAPIPE_METAL_ENABLED
MP_RETURN_IF_ERROR([MPPMetalHelper updateContract:cc]); MP_RETURN_IF_ERROR([MPPMetalHelper updateContract:cc]);