Utility functions that create RGB images for testing.

PiperOrigin-RevId: 730797962

Author: MediaPipe Team

mediapipe/util/BUILD

@@ -494,7 +494,6 @@ cc_library(
         "//mediapipe/framework/port:opencv_imgproc",
         "//mediapipe/gpu:gpu_buffer",
         "//mediapipe/gpu:gpu_buffer_format",
-        "//mediapipe/gpu:gpu_buffer_storage_image_frame",
         "@com_google_absl//absl/log:absl_log",
     ],
 )

mediapipe/util/image_test_utils.cc

@@ -116,6 +116,11 @@ ImageFrame CreateTestRgba8ImageFrame(int width, int height) {
       width, height, /*max_value=*/255.0f);
 }
 
+ImageFrame CreateTestRgb8ImageFrame(int width, int height) {
+  return CreateTestImageFrame<ImageFormat::SRGB, uint8_t>(width, height,
+                                                          /*max_value=*/255.0f);
+}
+
 GpuBuffer CreateTestFloat32GpuBuffer(int width, int height) {
   GpuBuffer buffer(width, height, GpuBufferFormat::kGrayFloat32);
   std::shared_ptr<ImageFrame> view = buffer.GetWriteView<ImageFrame>();
@@ -137,4 +142,11 @@ GpuBuffer CreateTestRgba8GpuBuffer(int width, int height) {
   return buffer;
 }
 
+GpuBuffer CreateTestRgb8GpuBuffer(int width, int height) {
+  GpuBuffer buffer(width, height, GpuBufferFormat::kRGB24);
+  std::shared_ptr<ImageFrame> view = buffer.GetWriteView<ImageFrame>();
+  *view = CreateTestRgb8ImageFrame(width, height);
+  return buffer;
+}
+
 }  // namespace mediapipe

mediapipe/util/image_test_utils.h

@@ -41,6 +41,9 @@ ImageFrame CreateTestGrey8ImageFrame(int width, int height);
 // Generates 4 channel uint8 RGBA ImageFrame with increasing [0,255] values.
 ImageFrame CreateTestRgba8ImageFrame(int width, int height);
 
+// Generates 3 channel uint8 RGB ImageFrame with increasing [0,255] values.
+ImageFrame CreateTestRgb8ImageFrame(int width, int height);
+
 // Generates single-channel float32 GpuBuffer with increasing [0,1] values.
 GpuBuffer CreateTestFloat32GpuBuffer(int width, int height);
 
@@ -50,6 +53,9 @@ GpuBuffer CreateTestGrey8GpuBuffer(int width, int height);
 // Generates 4 channel uint8 RGBA GpuBuffer with increasing [0,255] values.
 GpuBuffer CreateTestRgba8GpuBuffer(int width, int height);
 
+// Generates 3 channel uint8 RGB GpuBuffer with increasing [0,255] values.
+GpuBuffer CreateTestRgb8GpuBuffer(int width, int height);
+
 }  // namespace mediapipe
 
 #endif  // MEDIAPIPE_UTIL_IMAGE_TEST_UTILS_H_