add end loop calculator for image size
PiperOrigin-RevId: 556955370
This commit is contained in:
parent
a8bee6baf3
commit
6605fdb16f
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user