diff --git a/mediapipe/calculators/core/BUILD b/mediapipe/calculators/core/BUILD index d3e63e38..a425b7e3 100644 --- a/mediapipe/calculators/core/BUILD +++ b/mediapipe/calculators/core/BUILD @@ -289,6 +289,7 @@ cc_library( "//mediapipe/framework/api2:node", "//mediapipe/framework/api2:port", "//mediapipe/framework/formats:classification_cc_proto", + "//mediapipe/framework/formats:image", "//mediapipe/framework/formats:landmark_cc_proto", "//mediapipe/framework/formats:tensor", "//mediapipe/framework/port:integral_types", diff --git a/mediapipe/calculators/core/concatenate_vector_calculator.cc b/mediapipe/calculators/core/concatenate_vector_calculator.cc index 4d0d6620..53b3debf 100644 --- a/mediapipe/calculators/core/concatenate_vector_calculator.cc +++ b/mediapipe/calculators/core/concatenate_vector_calculator.cc @@ -17,6 +17,7 @@ #include #include "mediapipe/framework/formats/classification.pb.h" +#include "mediapipe/framework/formats/image.h" #include "mediapipe/framework/formats/landmark.pb.h" #include "mediapipe/framework/formats/tensor.h" #include "mediapipe/framework/port/integral_types.h" @@ -104,4 +105,7 @@ typedef ConcatenateVectorCalculator ConcatenateRenderDataVectorCalculator; MEDIAPIPE_REGISTER_NODE(ConcatenateRenderDataVectorCalculator); +typedef ConcatenateVectorCalculator + ConcatenateImageVectorCalculator; +MEDIAPIPE_REGISTER_NODE(ConcatenateImageVectorCalculator); } // namespace mediapipe