mirror of
https://github.com/opencv/opencv.git
synced 2024-12-05 09:49:12 +08:00
209 lines
6.9 KiB
C++
209 lines
6.9 KiB
C++
// This file is part of OpenCV project.
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
|
// of this distribution and at http://opencv.org/license.html.
|
|
|
|
#include "perf_precomp.hpp"
|
|
|
|
namespace opencv_test
|
|
{
|
|
namespace
|
|
{
|
|
|
|
typedef ::perf::TestBaseWithParam< std::string > Perf_Objdetect_QRCode;
|
|
|
|
PERF_TEST_P_(Perf_Objdetect_QRCode, detect)
|
|
{
|
|
const std::string name_current_image = GetParam();
|
|
const std::string root = "cv/qrcode/";
|
|
|
|
std::string image_path = findDataFile(root + name_current_image);
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
|
|
|
|
std::vector< Point > corners;
|
|
QRCodeDetector qrcode;
|
|
TEST_CYCLE() ASSERT_TRUE(qrcode.detect(src, corners));
|
|
SANITY_CHECK(corners);
|
|
}
|
|
|
|
#ifdef HAVE_QUIRC
|
|
PERF_TEST_P_(Perf_Objdetect_QRCode, decode)
|
|
{
|
|
const std::string name_current_image = GetParam();
|
|
const std::string root = "cv/qrcode/";
|
|
|
|
std::string image_path = findDataFile(root + name_current_image);
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
|
|
|
|
std::vector< Point > corners;
|
|
std::string decoded_info;
|
|
QRCodeDetector qrcode;
|
|
ASSERT_TRUE(qrcode.detect(src, corners));
|
|
TEST_CYCLE()
|
|
{
|
|
decoded_info = qrcode.decode(src, corners, straight_barcode);
|
|
ASSERT_FALSE(decoded_info.empty());
|
|
}
|
|
|
|
std::vector<uint8_t> decoded_info_uint8_t(decoded_info.begin(), decoded_info.end());
|
|
SANITY_CHECK(decoded_info_uint8_t);
|
|
SANITY_CHECK(straight_barcode);
|
|
|
|
}
|
|
#endif
|
|
|
|
typedef ::perf::TestBaseWithParam< std::string > Perf_Objdetect_QRCode_Multi;
|
|
|
|
static inline bool compareCorners(const Point2f& corner1, const Point2f& corner2) {
|
|
return corner1.x == corner2.x ? corner1.y < corner2.y : corner1.x < corner2.x;
|
|
}
|
|
|
|
PERF_TEST_P_(Perf_Objdetect_QRCode_Multi, detectMulti)
|
|
{
|
|
const std::string name_current_image = GetParam();
|
|
const std::string root = "cv/qrcode/multiple/";
|
|
|
|
std::string image_path = findDataFile(root + name_current_image);
|
|
Mat src = imread(image_path);
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
|
|
std::vector<Point2f> corners;
|
|
QRCodeDetector qrcode;
|
|
TEST_CYCLE() ASSERT_TRUE(qrcode.detectMulti(src, corners));
|
|
sort(corners.begin(), corners.end(), compareCorners);
|
|
SANITY_CHECK(corners);
|
|
}
|
|
|
|
static inline bool compareQR(const pair<string, Mat>& v1, const pair<string, Mat>& v2) {
|
|
return v1.first < v2.first;
|
|
}
|
|
|
|
#ifdef HAVE_QUIRC
|
|
PERF_TEST_P_(Perf_Objdetect_QRCode_Multi, decodeMulti)
|
|
{
|
|
const std::string name_current_image = GetParam();
|
|
const std::string root = "cv/qrcode/multiple/";
|
|
std::string image_path = findDataFile(root + name_current_image);
|
|
Mat src = imread(image_path);
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
|
|
QRCodeDetector qrcode;
|
|
std::vector<Point2f> corners;
|
|
ASSERT_TRUE(qrcode.detectMulti(src, corners));
|
|
std::vector<Mat> straight_barcode;
|
|
std::vector< cv::String > decoded_info;
|
|
TEST_CYCLE()
|
|
{
|
|
ASSERT_TRUE(qrcode.decodeMulti(src, corners, decoded_info, straight_barcode));
|
|
for(size_t i = 0; i < decoded_info.size(); i++)
|
|
{
|
|
ASSERT_FALSE(decoded_info[i].empty());
|
|
}
|
|
}
|
|
ASSERT_EQ(decoded_info.size(), straight_barcode.size());
|
|
vector<pair<string, Mat> > result;
|
|
for (size_t i = 0ull; i < decoded_info.size(); i++) {
|
|
result.push_back(make_pair(decoded_info[i], straight_barcode[i]));
|
|
}
|
|
|
|
sort(result.begin(), result.end(), compareQR);
|
|
vector<vector<uint8_t> > decoded_info_sort;
|
|
vector<Mat> straight_barcode_sort;
|
|
for (size_t i = 0ull; i < result.size(); i++) {
|
|
vector<uint8_t> tmp(result[i].first.begin(), result[i].first.end());
|
|
decoded_info_sort.push_back(tmp);
|
|
straight_barcode_sort.push_back(result[i].second);
|
|
}
|
|
SANITY_CHECK(decoded_info_sort);
|
|
}
|
|
#endif
|
|
|
|
INSTANTIATE_TEST_CASE_P(/*nothing*/, Perf_Objdetect_QRCode,
|
|
::testing::Values(
|
|
"version_1_down.jpg", "version_1_left.jpg", "version_1_right.jpg", "version_1_up.jpg", "version_1_top.jpg",
|
|
"version_5_down.jpg", "version_5_left.jpg",/*version_5_right.jpg*/ "version_5_up.jpg", "version_5_top.jpg",
|
|
"russian.jpg", "kanji.jpg", "link_github_ocv.jpg", "link_ocv.jpg", "link_wiki_cv.jpg"
|
|
)
|
|
);
|
|
// version_5_right.jpg DISABLED after tile fix, PR #22025
|
|
|
|
INSTANTIATE_TEST_CASE_P(/*nothing*/, Perf_Objdetect_QRCode_Multi,
|
|
::testing::Values(
|
|
"2_qrcodes.png", "3_close_qrcodes.png", "3_qrcodes.png", "4_qrcodes.png",
|
|
"5_qrcodes.png", "6_qrcodes.png", "7_qrcodes.png", "8_close_qrcodes.png"
|
|
)
|
|
);
|
|
|
|
typedef ::perf::TestBaseWithParam< tuple< std::string, Size > > Perf_Objdetect_Not_QRCode;
|
|
|
|
PERF_TEST_P_(Perf_Objdetect_Not_QRCode, detect)
|
|
{
|
|
std::vector<Point> corners;
|
|
std::string type_gen = get<0>(GetParam());
|
|
Size resolution = get<1>(GetParam());
|
|
Mat not_qr_code(resolution, CV_8UC1, Scalar(0));
|
|
if (type_gen == "random")
|
|
{
|
|
RNG rng;
|
|
rng.fill(not_qr_code, RNG::UNIFORM, Scalar(0), Scalar(1));
|
|
}
|
|
if (type_gen == "chessboard")
|
|
{
|
|
uint8_t next_pixel = 0;
|
|
for (int r = 0; r < not_qr_code.rows * not_qr_code.cols; r++)
|
|
{
|
|
int i = r / not_qr_code.cols;
|
|
int j = r % not_qr_code.cols;
|
|
not_qr_code.ptr<uchar>(i)[j] = next_pixel;
|
|
next_pixel = 255 - next_pixel;
|
|
}
|
|
}
|
|
|
|
QRCodeDetector qrcode;
|
|
TEST_CYCLE() ASSERT_FALSE(qrcode.detect(not_qr_code, corners));
|
|
SANITY_CHECK_NOTHING();
|
|
}
|
|
|
|
#ifdef HAVE_QUIRC
|
|
PERF_TEST_P_(Perf_Objdetect_Not_QRCode, decode)
|
|
{
|
|
Mat straight_barcode;
|
|
std::vector< Point > corners;
|
|
corners.push_back(Point( 0, 0)); corners.push_back(Point( 0, 5));
|
|
corners.push_back(Point(10, 0)); corners.push_back(Point(15, 15));
|
|
|
|
std::string type_gen = get<0>(GetParam());
|
|
Size resolution = get<1>(GetParam());
|
|
Mat not_qr_code(resolution, CV_8UC1, Scalar(0));
|
|
if (type_gen == "random")
|
|
{
|
|
RNG rng;
|
|
rng.fill(not_qr_code, RNG::UNIFORM, Scalar(0), Scalar(1));
|
|
}
|
|
if (type_gen == "chessboard")
|
|
{
|
|
uint8_t next_pixel = 0;
|
|
for (int r = 0; r < not_qr_code.rows * not_qr_code.cols; r++)
|
|
{
|
|
int i = r / not_qr_code.cols;
|
|
int j = r % not_qr_code.cols;
|
|
not_qr_code.ptr<uchar>(i)[j] = next_pixel;
|
|
next_pixel = 255 - next_pixel;
|
|
}
|
|
}
|
|
|
|
QRCodeDetector qrcode;
|
|
TEST_CYCLE() ASSERT_TRUE(qrcode.decode(not_qr_code, corners, straight_barcode).empty());
|
|
SANITY_CHECK_NOTHING();
|
|
}
|
|
#endif
|
|
|
|
INSTANTIATE_TEST_CASE_P(/*nothing*/, Perf_Objdetect_Not_QRCode,
|
|
::testing::Combine(
|
|
::testing::Values("zero", "random", "chessboard"),
|
|
::testing::Values(Size(640, 480), Size(1280, 720),
|
|
Size(1920, 1080), Size(3840, 2160))
|
|
));
|
|
|
|
}
|
|
} // namespace
|