add end loop calculator for image size

PiperOrigin-RevId: 556955370
This commit is contained in:
MediaPipe Team 2023-08-14 17:05:49 -07:00 committed by Copybara-Service
parent a8bee6baf3
commit 6605fdb16f

View File

@ -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