diff --git a/mediapipe/calculators/core/end_loop_calculator.cc b/mediapipe/calculators/core/end_loop_calculator.cc index 752580cfd..94f7ee22e 100644 --- a/mediapipe/calculators/core/end_loop_calculator.cc +++ b/mediapipe/calculators/core/end_loop_calculator.cc @@ -14,6 +14,8 @@ #include "mediapipe/calculators/core/end_loop_calculator.h" +#include +#include #include #include "mediapipe/framework/formats/classification.pb.h" @@ -84,4 +86,8 @@ typedef EndLoopCalculator>> EndLoopAffineMatrixCalculator; REGISTER_CALCULATOR(EndLoopAffineMatrixCalculator); +typedef EndLoopCalculator>> + EndLoopImageSizeCalculator; +REGISTER_CALCULATOR(EndLoopImageSizeCalculator); + } // namespace mediapipe