|
|
|
|
@ -15,6 +15,12 @@ std::string qrcode_images_name[] = { |
|
|
|
|
"russian.jpg", "kanji.jpg", "link_github_ocv.jpg", "link_ocv.jpg", "link_wiki_cv.jpg" |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
std::string qrcode_images_close[] = { |
|
|
|
|
"close_1.png", "close_2.png", "close_3.png", "close_4.png", "close_5.png" |
|
|
|
|
}; |
|
|
|
|
std::string qrcode_images_monitor[] = { |
|
|
|
|
"monitor_1.png", "monitor_2.png", "monitor_3.png", "monitor_4.png", "monitor_5.png" |
|
|
|
|
}; |
|
|
|
|
// #define UPDATE_QRCODE_TEST_DATA
|
|
|
|
|
#ifdef UPDATE_QRCODE_TEST_DATA |
|
|
|
|
|
|
|
|
|
@ -51,6 +57,83 @@ TEST(Objdetect_QRCode, generate_test_data) |
|
|
|
|
file_config.release(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
TEST(Objdetect_QRCode_Close, generate_test_data) |
|
|
|
|
{ |
|
|
|
|
const std::string root = "qrcode/close/"; |
|
|
|
|
const std::string dataset_config = findDataFile(root + "dataset_config.json"); |
|
|
|
|
FileStorage file_config(dataset_config, FileStorage::WRITE); |
|
|
|
|
|
|
|
|
|
file_config << "close_images" << "["; |
|
|
|
|
size_t close_count = sizeof(qrcode_images_close) / sizeof(qrcode_images_close[0]); |
|
|
|
|
for (size_t i = 0; i < close_count; i++) |
|
|
|
|
{ |
|
|
|
|
file_config << "{:" << "image_name" << qrcode_images_close[i]; |
|
|
|
|
std::string image_path = findDataFile(root + qrcode_images_close[i]); |
|
|
|
|
std::vector<Point> corners; |
|
|
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode; |
|
|
|
|
std::string decoded_info; |
|
|
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path; |
|
|
|
|
const double min_side = std::min(src.size().width, src.size().height); |
|
|
|
|
double coeff_expansion = 1024.0 / min_side; |
|
|
|
|
const int width = cvRound(src.size().width * coeff_expansion); |
|
|
|
|
const int height = cvRound(src.size().height * coeff_expansion); |
|
|
|
|
Size new_size(width, height); |
|
|
|
|
resize(src, barcode, new_size, 0, 0, INTER_LINEAR); |
|
|
|
|
EXPECT_TRUE(detectQRCode(barcode, corners)); |
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
EXPECT_TRUE(decodeQRCode(barcode, corners, decoded_info, straight_barcode)); |
|
|
|
|
#endif |
|
|
|
|
file_config << "x" << "[:"; |
|
|
|
|
for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; } |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config << "y" << "[:"; |
|
|
|
|
for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; } |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config << "info" << decoded_info; |
|
|
|
|
file_config << "}"; |
|
|
|
|
} |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config.release(); |
|
|
|
|
} |
|
|
|
|
TEST(Objdetect_QRCode_Monitor, generate_test_data) |
|
|
|
|
{ |
|
|
|
|
const std::string root = "qrcode/monitor/"; |
|
|
|
|
const std::string dataset_config = findDataFile(root + "dataset_config.json"); |
|
|
|
|
FileStorage file_config(dataset_config, FileStorage::WRITE); |
|
|
|
|
|
|
|
|
|
file_config << "monitor_images" << "["; |
|
|
|
|
size_t monitor_count = sizeof(qrcode_images_monitor) / sizeof(qrcode_images_monitor[0]); |
|
|
|
|
for (size_t i = 0; i < monitor_count; i++) |
|
|
|
|
{ |
|
|
|
|
file_config << "{:" << "image_name" << qrcode_images_monitor[i]; |
|
|
|
|
std::string image_path = findDataFile(root + qrcode_images_monitor[i]); |
|
|
|
|
std::vector<Point> corners; |
|
|
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode; |
|
|
|
|
std::string decoded_info; |
|
|
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path; |
|
|
|
|
const double min_side = std::min(src.size().width, src.size().height); |
|
|
|
|
double coeff_expansion = 1024.0 / min_side; |
|
|
|
|
const int width = cvRound(src.size().width * coeff_expansion); |
|
|
|
|
const int height = cvRound(src.size().height * coeff_expansion); |
|
|
|
|
Size new_size(width, height); |
|
|
|
|
resize(src, barcode, new_size, 0, 0, INTER_LINEAR); |
|
|
|
|
EXPECT_TRUE(detectQRCode(barcode, corners)); |
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
EXPECT_TRUE(decodeQRCode(barcode, corners, decoded_info, straight_barcode)); |
|
|
|
|
#endif |
|
|
|
|
file_config << "x" << "[:"; |
|
|
|
|
for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; } |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config << "y" << "[:"; |
|
|
|
|
for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; } |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config << "info" << decoded_info; |
|
|
|
|
file_config << "}"; |
|
|
|
|
} |
|
|
|
|
file_config << "]"; |
|
|
|
|
file_config.release(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#else |
|
|
|
|
|
|
|
|
|
typedef testing::TestWithParam< std::string > Objdetect_QRCode; |
|
|
|
|
@ -113,9 +196,139 @@ TEST_P(Objdetect_QRCode, regression) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode, testing::ValuesIn(qrcode_images_name)); |
|
|
|
|
typedef testing::TestWithParam< std::string > Objdetect_QRCode_Close; |
|
|
|
|
TEST_P(Objdetect_QRCode_Close, regression) |
|
|
|
|
{ |
|
|
|
|
const std::string name_current_image = GetParam(); |
|
|
|
|
const std::string root = "qrcode/close/"; |
|
|
|
|
const int pixels_error = 3; |
|
|
|
|
|
|
|
|
|
std::string image_path = findDataFile(root + name_current_image); |
|
|
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode; |
|
|
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path; |
|
|
|
|
const double min_side = std::min(src.size().width, src.size().height); |
|
|
|
|
double coeff_expansion = 1024.0 / min_side; |
|
|
|
|
const int width = cvRound(src.size().width * coeff_expansion); |
|
|
|
|
const int height = cvRound(src.size().height * coeff_expansion); |
|
|
|
|
Size new_size(width, height); |
|
|
|
|
resize(src, barcode, new_size, 0, 0, INTER_LINEAR); |
|
|
|
|
std::vector<Point> corners; |
|
|
|
|
std::string decoded_info; |
|
|
|
|
QRCodeDetector qrcode; |
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
decoded_info = qrcode.detectAndDecode(barcode, corners, straight_barcode); |
|
|
|
|
ASSERT_FALSE(corners.empty()); |
|
|
|
|
ASSERT_FALSE(decoded_info.empty()); |
|
|
|
|
#else |
|
|
|
|
ASSERT_TRUE(qrcode.detect(src, corners)); |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
const std::string dataset_config = findDataFile(root + "dataset_config.json"); |
|
|
|
|
FileStorage file_config(dataset_config, FileStorage::READ); |
|
|
|
|
ASSERT_TRUE(file_config.isOpened()) << "Can't read validation data: " << dataset_config; |
|
|
|
|
{ |
|
|
|
|
FileNode images_list = file_config["close_images"]; |
|
|
|
|
size_t images_count = static_cast<size_t>(images_list.size()); |
|
|
|
|
ASSERT_GT(images_count, 0u) << "Can't find validation data entries in 'test_images': " << dataset_config; |
|
|
|
|
|
|
|
|
|
for (size_t index = 0; index < images_count; index++) |
|
|
|
|
{ |
|
|
|
|
FileNode config = images_list[(int)index]; |
|
|
|
|
std::string name_test_image = config["image_name"]; |
|
|
|
|
if (name_test_image == name_current_image) |
|
|
|
|
{ |
|
|
|
|
for (int i = 0; i < 4; i++) |
|
|
|
|
{ |
|
|
|
|
int x = config["x"][i]; |
|
|
|
|
int y = config["y"][i]; |
|
|
|
|
EXPECT_NEAR(x, corners[i].x, pixels_error); |
|
|
|
|
EXPECT_NEAR(y, corners[i].y, pixels_error); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
std::string original_info = config["info"]; |
|
|
|
|
EXPECT_EQ(decoded_info, original_info); |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
return; // done
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
std::cerr |
|
|
|
|
<< "Not found results for '" << name_current_image |
|
|
|
|
<< "' image in config file:" << dataset_config << std::endl |
|
|
|
|
<< "Re-run tests with enabled UPDATE_QRCODE_TEST_DATA macro to update test data." |
|
|
|
|
<< std::endl; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
typedef testing::TestWithParam< std::string > Objdetect_QRCode_Monitor; |
|
|
|
|
TEST_P(Objdetect_QRCode_Monitor, regression) |
|
|
|
|
{ |
|
|
|
|
const std::string name_current_image = GetParam(); |
|
|
|
|
const std::string root = "qrcode/monitor/"; |
|
|
|
|
const int pixels_error = 3; |
|
|
|
|
|
|
|
|
|
std::string image_path = findDataFile(root + name_current_image); |
|
|
|
|
Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode; |
|
|
|
|
ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path; |
|
|
|
|
const double min_side = std::min(src.size().width, src.size().height); |
|
|
|
|
double coeff_expansion = 1024.0 / min_side; |
|
|
|
|
const int width = cvRound(src.size().width * coeff_expansion); |
|
|
|
|
const int height = cvRound(src.size().height * coeff_expansion); |
|
|
|
|
Size new_size(width, height); |
|
|
|
|
resize(src, barcode, new_size, 0, 0, INTER_LINEAR); |
|
|
|
|
std::vector<Point> corners; |
|
|
|
|
std::string decoded_info; |
|
|
|
|
QRCodeDetector qrcode; |
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
decoded_info = qrcode.detectAndDecode(barcode, corners, straight_barcode); |
|
|
|
|
ASSERT_FALSE(corners.empty()); |
|
|
|
|
ASSERT_FALSE(decoded_info.empty()); |
|
|
|
|
#else |
|
|
|
|
ASSERT_TRUE(qrcode.detect(src, corners)); |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
const std::string dataset_config = findDataFile(root + "dataset_config.json"); |
|
|
|
|
FileStorage file_config(dataset_config, FileStorage::READ); |
|
|
|
|
ASSERT_TRUE(file_config.isOpened()) << "Can't read validation data: " << dataset_config; |
|
|
|
|
{ |
|
|
|
|
FileNode images_list = file_config["monitor_images"]; |
|
|
|
|
size_t images_count = static_cast<size_t>(images_list.size()); |
|
|
|
|
ASSERT_GT(images_count, 0u) << "Can't find validation data entries in 'test_images': " << dataset_config; |
|
|
|
|
|
|
|
|
|
for (size_t index = 0; index < images_count; index++) |
|
|
|
|
{ |
|
|
|
|
FileNode config = images_list[(int)index]; |
|
|
|
|
std::string name_test_image = config["image_name"]; |
|
|
|
|
if (name_test_image == name_current_image) |
|
|
|
|
{ |
|
|
|
|
for (int i = 0; i < 4; i++) |
|
|
|
|
{ |
|
|
|
|
int x = config["x"][i]; |
|
|
|
|
int y = config["y"][i]; |
|
|
|
|
EXPECT_NEAR(x, corners[i].x, pixels_error); |
|
|
|
|
EXPECT_NEAR(y, corners[i].y, pixels_error); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#ifdef HAVE_QUIRC |
|
|
|
|
std::string original_info = config["info"]; |
|
|
|
|
EXPECT_EQ(decoded_info, original_info); |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
return; // done
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
std::cerr |
|
|
|
|
<< "Not found results for '" << name_current_image |
|
|
|
|
<< "' image in config file:" << dataset_config << std::endl |
|
|
|
|
<< "Re-run tests with enabled UPDATE_QRCODE_TEST_DATA macro to update test data." |
|
|
|
|
<< std::endl; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode, testing::ValuesIn(qrcode_images_name)); |
|
|
|
|
INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Close, testing::ValuesIn(qrcode_images_close)); |
|
|
|
|
INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Monitor, testing::ValuesIn(qrcode_images_monitor)); |
|
|
|
|
|
|
|
|
|
TEST(Objdetect_QRCode_basic, not_found_qrcode) |
|
|
|
|
{ |
|
|
|
|
|
Alexander Alekhin
6 years ago