From 6605fdb16f039586466aea0cb04ac5bcdf5b6138 Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Mon, 14 Aug 2023 17:05:49 -0700
Subject: [PATCH] add end loop calculator for image size

PiperOrigin-RevId: 556955370
---
 mediapipe/calculators/core/end_loop_calculator.cc | 6 ++++++
 1 file changed, 6 insertions(+)

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 <array>
+#include <utility>
 #include <vector>
 
 #include "mediapipe/framework/formats/classification.pb.h"
@@ -84,4 +86,8 @@ typedef EndLoopCalculator<std::vector<std::array<float, 16>>>
     EndLoopAffineMatrixCalculator;
 REGISTER_CALCULATOR(EndLoopAffineMatrixCalculator);
 
+typedef EndLoopCalculator<std::vector<std::pair<int, int>>>
+    EndLoopImageSizeCalculator;
+REGISTER_CALCULATOR(EndLoopImageSizeCalculator);
+
 }  // namespace mediapipe