add end loop calculator for image size
PiperOrigin-RevId: 556955370
This commit is contained in:
committed by
Copybara-Service
parent
a8bee6baf3
commit
6605fdb16f
@@ -14,6 +14,8 @@
|
|||||||
|
|
||||||
#include "mediapipe/calculators/core/end_loop_calculator.h"
|
#include "mediapipe/calculators/core/end_loop_calculator.h"
|
||||||
|
|
||||||
|
#include <array>
|
||||||
|
#include <utility>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "mediapipe/framework/formats/classification.pb.h"
|
#include "mediapipe/framework/formats/classification.pb.h"
|
||||||
@@ -84,4 +86,8 @@ typedef EndLoopCalculator<std::vector<std::array<float, 16>>>
|
|||||||
EndLoopAffineMatrixCalculator;
|
EndLoopAffineMatrixCalculator;
|
||||||
REGISTER_CALCULATOR(EndLoopAffineMatrixCalculator);
|
REGISTER_CALCULATOR(EndLoopAffineMatrixCalculator);
|
||||||
|
|
||||||
|
typedef EndLoopCalculator<std::vector<std::pair<int, int>>>
|
||||||
|
EndLoopImageSizeCalculator;
|
||||||
|
REGISTER_CALCULATOR(EndLoopImageSizeCalculator);
|
||||||
|
|
||||||
} // namespace mediapipe
|
} // namespace mediapipe
|
||||||
|
|||||||
Reference in New Issue
Block a user