Function TEST

oneflow/user/image/jpeg_decoder_test.cpp:64–94

}

TEST(JPEG, decoder) {
  constexpr size_t test_num = 3;
  std::vector<unsigned char> jpg;
  GenerateImage(jpg, 192, 192);
  std::seed_seq seq{1, 2, 3};
  std::vector<int64_t> seeds(test_num);
  seq.generate(seeds.begin(), seeds.end());

  for (int i = 0; i < test_num; i++) {
    cv::Mat libjpeg_image_mat;

    RandomCropGenerator libjpeg_random_crop_gen({0.1, 0.9}, {0.4, 0.6}, seeds[i], 1);
    RandomCropGenerator opencv_random_crop_gen({0.1, 0.9}, {0.4, 0.6}, seeds[i], 1);
    auto status = JpegPartialDecodeRandomCropImage(jpg.data(), jpg.size(), &libjpeg_random_crop_gen,
                                                   nullptr, 0, &libjpeg_image_mat);
    ASSERT_EQ(status, true);

    cv::Mat opencv_image_mat;
    std::string color_space("RGB");

    OpenCvPartialDecodeRandomCropImage(jpg.data(), jpg.size(), &opencv_random_crop_gen, color_space,
                                       opencv_image_mat);
    ImageUtil::ConvertColor("BGR", opencv_image_mat, color_space, opencv_image_mat);

    cv::Mat checkout = libjpeg_image_mat - opencv_image_mat;
    auto sum = cv::sum(cv::sum(checkout));
    ASSERT_EQ(sum[0], 0);
    // cv::imwrite("jpeg.ppm", libjpeg_image_mat);
    // cv::imwrite("opencv.ppm", opencv_image_mat);
  }
}

}  // namespace oneflow