Internal change
PiperOrigin-RevId: 516179167
| 
						 | 
				
			
			@ -138,7 +138,23 @@ void TestWithAspectRatio(const double aspect_ratio,
 | 
			
		|||
    std::string result_image;
 | 
			
		||||
    MP_ASSERT_OK(
 | 
			
		||||
        mediapipe::file::GetContents(result_string_path, &result_image));
 | 
			
		||||
    EXPECT_EQ(result_image, output_string);
 | 
			
		||||
    if (result_image != output_string) {
 | 
			
		||||
      // There may be slight differences due to the way the JPEG was encoded or
 | 
			
		||||
      // the OpenCV version used to generate the reference files. Compare
 | 
			
		||||
      // pixel-by-pixel using the Peak Signal-to-Noise Ratio instead.
 | 
			
		||||
      cv::Mat result_mat =
 | 
			
		||||
          cv::imdecode(cv::Mat(1, result_image.size(), CV_8UC1,
 | 
			
		||||
                               const_cast<char*>(result_image.data())),
 | 
			
		||||
                       cv::IMREAD_UNCHANGED);
 | 
			
		||||
      cv::Mat output_mat =
 | 
			
		||||
          cv::imdecode(cv::Mat(1, output_string.size(), CV_8UC1,
 | 
			
		||||
                               const_cast<char*>(output_string.data())),
 | 
			
		||||
                       cv::IMREAD_UNCHANGED);
 | 
			
		||||
      ASSERT_EQ(result_mat.rows, output_mat.rows);
 | 
			
		||||
      ASSERT_EQ(result_mat.cols, output_mat.cols);
 | 
			
		||||
      ASSERT_EQ(result_mat.type(), output_mat.type());
 | 
			
		||||
      EXPECT_GT(cv::PSNR(result_mat, output_mat), 45.0);
 | 
			
		||||
    }
 | 
			
		||||
  } else {
 | 
			
		||||
    std::string output_string_path = mediapipe::file::JoinPath(
 | 
			
		||||
        absl::GetFlag(FLAGS_output_folder),
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
		 Before Width: | Height: | Size: 3.2 KiB After Width: | Height: | Size: 3.2 KiB  | 
| 
		 Before Width: | Height: | Size: 6.1 KiB After Width: | Height: | Size: 6.1 KiB  | 
| 
		 Before Width: | Height: | Size: 8.2 KiB After Width: | Height: | Size: 8.2 KiB  | 
| 
		 Before Width: | Height: | Size: 7.6 KiB After Width: | Height: | Size: 7.6 KiB  |