Extracted common test helper functions out from the unittest into a sharable
library. Also migrated away from OpenCVX. PiperOrigin-RevId: 490074410
This commit is contained in:
committed by
Copybara-Service
parent
652423a23d
commit
adddf2c2ab
@@ -0,0 +1,32 @@
|
||||
#ifndef MEDIAPIPE_UTIL_IMAGE_TEST_UTILS_H_
|
||||
#define MEDIAPIPE_UTIL_IMAGE_TEST_UTILS_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "mediapipe/framework/formats/image.h"
|
||||
#include "mediapipe/framework/packet.h"
|
||||
#include "mediapipe/framework/port/opencv_core_inc.h"
|
||||
|
||||
namespace mediapipe {
|
||||
|
||||
// Reads the image file into cv::Mat with RGB channels.
|
||||
cv::Mat GetRgb(const std::string& path);
|
||||
|
||||
// Reads the image file into cv::Mat with RGBA channels.
|
||||
cv::Mat GetRgba(const std::string& path);
|
||||
|
||||
// Reads the image file into cv::Mat with Gray channel.
|
||||
cv::Mat GetGray(const std::string& path);
|
||||
|
||||
// Converts the image channels into corresponding ImageFormat.
|
||||
mediapipe::ImageFormat::Format GetImageFormat(int image_channels);
|
||||
|
||||
// Converts the cv::Mat into ImageFrame packet.
|
||||
Packet MakeImageFramePacket(cv::Mat input, int timestamp = 0);
|
||||
|
||||
// Converts the cv::Mat into Image packet.
|
||||
Packet MakeImagePacket(cv::Mat input, int timestamp = 0);
|
||||
|
||||
} // namespace mediapipe
|
||||
|
||||
#endif // MEDIAPIPE_UTIL_IMAGE_TEST_UTILS_H_
|
||||
Reference in New Issue
Block a user