test_qrcode.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  1. // This file is part of OpenCV project.
  2. // It is subject to the license terms in the LICENSE file found in the top-level directory
  3. // of this distribution and at http://opencv.org/license.html.
  4. #include "test_precomp.hpp"
  5. #include "test_qr_utils.hpp"
  6. #include "opencv2/imgproc.hpp"
  7. namespace opencv_test { namespace {
  8. std::string qrcode_images_name[] = {
  9. "version_1_down.jpg", "version_1_left.jpg", "version_1_right.jpg", "version_1_up.jpg", "version_1_top.jpg",
  10. "version_2_down.jpg", "version_2_left.jpg", "version_2_right.jpg", "version_2_up.jpg", "version_2_top.jpg",
  11. "version_3_down.jpg", "version_3_left.jpg", "version_3_right.jpg", "version_3_up.jpg", "version_3_top.jpg",
  12. "version_4_down.jpg", "version_4_left.jpg", "version_4_right.jpg", "version_4_up.jpg", "version_4_top.jpg",
  13. "version_5_down.jpg", "version_5_left.jpg", /*"version_5_right.jpg",*/ "version_5_up.jpg", "version_5_top.jpg",
  14. "russian.jpg", "kanji.jpg", "link_github_ocv.jpg", "link_ocv.jpg", "link_wiki_cv.jpg"
  15. // version_5_right.jpg DISABLED after tile fix, PR #22025
  16. };
  17. // Todo: fix corner align in big QRs to enable close_5.png
  18. std::string qrcode_images_close[] = {
  19. "close_1.png", "close_2.png", "close_3.png", "close_4.png"//, "close_5.png"
  20. };
  21. std::string qrcode_images_monitor[] = {
  22. "monitor_1.png", "monitor_2.png", "monitor_3.png", "monitor_4.png", "monitor_5.png"
  23. };
  24. std::string qrcode_images_curved[] = {
  25. "curved_1.jpg", "curved_2.jpg", "curved_3.jpg", /*"curved_4.jpg",*/ "curved_5.jpg", /*"curved_6.jpg",*/ "curved_7.jpg", "curved_8.jpg"
  26. };
  27. // curved_4.jpg, curved_6.jpg DISABLED after tile fix, PR #22025
  28. std::string qrcode_images_multiple[] = {
  29. "2_qrcodes.png", "3_close_qrcodes.png", "3_qrcodes.png", "4_qrcodes.png",
  30. "5_qrcodes.png", "6_qrcodes.png", "7_qrcodes.png", "8_close_qrcodes.png"
  31. };
  32. static std::set<std::pair<std::string, std::string>> disabled_samples = {{"5_qrcodes.png", "aruco_based"}};
  33. //#define UPDATE_QRCODE_TEST_DATA
  34. #ifdef UPDATE_QRCODE_TEST_DATA
  35. TEST(Objdetect_QRCode, generate_test_data)
  36. {
  37. const std::string root = "qrcode/";
  38. const std::string dataset_config = findDataFile(root + "dataset_config.json");
  39. FileStorage file_config(dataset_config, FileStorage::WRITE);
  40. file_config << "test_images" << "[";
  41. size_t images_count = sizeof(qrcode_images_name) / sizeof(qrcode_images_name[0]);
  42. for (size_t i = 0; i < images_count; i++)
  43. {
  44. file_config << "{:" << "image_name" << qrcode_images_name[i];
  45. std::string image_path = findDataFile(root + qrcode_images_name[i]);
  46. std::vector<Point> corners;
  47. Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
  48. std::string decoded_info;
  49. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  50. EXPECT_TRUE(detectQRCode(src, corners));
  51. EXPECT_TRUE(decodeQRCode(src, corners, decoded_info, straight_barcode));
  52. file_config << "x" << "[:";
  53. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; }
  54. file_config << "]";
  55. file_config << "y" << "[:";
  56. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; }
  57. file_config << "]";
  58. file_config << "info" << decoded_info;
  59. file_config << "}";
  60. }
  61. file_config << "]";
  62. file_config.release();
  63. }
  64. TEST(Objdetect_QRCode_Close, generate_test_data)
  65. {
  66. const std::string root = "qrcode/close/";
  67. const std::string dataset_config = findDataFile(root + "dataset_config.json");
  68. FileStorage file_config(dataset_config, FileStorage::WRITE);
  69. file_config << "close_images" << "[";
  70. size_t close_count = sizeof(qrcode_images_close) / sizeof(qrcode_images_close[0]);
  71. for (size_t i = 0; i < close_count; i++)
  72. {
  73. file_config << "{:" << "image_name" << qrcode_images_close[i];
  74. std::string image_path = findDataFile(root + qrcode_images_close[i]);
  75. std::vector<Point> corners;
  76. Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode;
  77. std::string decoded_info;
  78. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  79. const double min_side = std::min(src.size().width, src.size().height);
  80. double coeff_expansion = 1024.0 / min_side;
  81. const int width = cvRound(src.size().width * coeff_expansion);
  82. const int height = cvRound(src.size().height * coeff_expansion);
  83. Size new_size(width, height);
  84. resize(src, barcode, new_size, 0, 0, INTER_LINEAR_EXACT);
  85. EXPECT_TRUE(detectQRCode(barcode, corners));
  86. EXPECT_TRUE(decodeQRCode(barcode, corners, decoded_info, straight_barcode));
  87. file_config << "x" << "[:";
  88. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; }
  89. file_config << "]";
  90. file_config << "y" << "[:";
  91. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; }
  92. file_config << "]";
  93. file_config << "info" << decoded_info;
  94. file_config << "}";
  95. }
  96. file_config << "]";
  97. file_config.release();
  98. }
  99. TEST(Objdetect_QRCode_Monitor, generate_test_data)
  100. {
  101. const std::string root = "qrcode/monitor/";
  102. const std::string dataset_config = findDataFile(root + "dataset_config.json");
  103. FileStorage file_config(dataset_config, FileStorage::WRITE);
  104. file_config << "monitor_images" << "[";
  105. size_t monitor_count = sizeof(qrcode_images_monitor) / sizeof(qrcode_images_monitor[0]);
  106. for (size_t i = 0; i < monitor_count; i++)
  107. {
  108. file_config << "{:" << "image_name" << qrcode_images_monitor[i];
  109. std::string image_path = findDataFile(root + qrcode_images_monitor[i]);
  110. std::vector<Point> corners;
  111. Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode;
  112. std::string decoded_info;
  113. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  114. const double min_side = std::min(src.size().width, src.size().height);
  115. double coeff_expansion = 1024.0 / min_side;
  116. const int width = cvRound(src.size().width * coeff_expansion);
  117. const int height = cvRound(src.size().height * coeff_expansion);
  118. Size new_size(width, height);
  119. resize(src, barcode, new_size, 0, 0, INTER_LINEAR_EXACT);
  120. EXPECT_TRUE(detectQRCode(barcode, corners));
  121. EXPECT_TRUE(decodeQRCode(barcode, corners, decoded_info, straight_barcode));
  122. file_config << "x" << "[:";
  123. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; }
  124. file_config << "]";
  125. file_config << "y" << "[:";
  126. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; }
  127. file_config << "]";
  128. file_config << "info" << decoded_info;
  129. file_config << "}";
  130. }
  131. file_config << "]";
  132. file_config.release();
  133. }
  134. TEST(Objdetect_QRCode_Curved, generate_test_data)
  135. {
  136. const std::string root = "qrcode/curved/";
  137. const std::string dataset_config = findDataFile(root + "dataset_config.json");
  138. FileStorage file_config(dataset_config, FileStorage::WRITE);
  139. file_config << "test_images" << "[";
  140. size_t images_count = sizeof(qrcode_images_curved) / sizeof(qrcode_images_curved[0]);
  141. for (size_t i = 0; i < images_count; i++)
  142. {
  143. file_config << "{:" << "image_name" << qrcode_images_curved[i];
  144. std::string image_path = findDataFile(root + qrcode_images_curved[i]);
  145. std::vector<Point> corners;
  146. Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
  147. std::string decoded_info;
  148. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  149. EXPECT_TRUE(detectQRCode(src, corners));
  150. EXPECT_TRUE(decodeCurvedQRCode(src, corners, decoded_info, straight_barcode));
  151. file_config << "x" << "[:";
  152. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].x; }
  153. file_config << "]";
  154. file_config << "y" << "[:";
  155. for (size_t j = 0; j < corners.size(); j++) { file_config << corners[j].y; }
  156. file_config << "]";
  157. file_config << "info" << decoded_info;
  158. file_config << "}";
  159. }
  160. file_config << "]";
  161. file_config.release();
  162. }
  163. TEST(Objdetect_QRCode_Multi, generate_test_data)
  164. {
  165. const std::string root = "qrcode/multiple/";
  166. const std::string dataset_config = findDataFile(root + "dataset_config.json");
  167. FileStorage file_config(dataset_config, FileStorage::WRITE);
  168. file_config << "multiple_images" << "[:";
  169. size_t multiple_count = sizeof(qrcode_images_multiple) / sizeof(qrcode_images_multiple[0]);
  170. for (size_t i = 0; i < multiple_count; i++)
  171. {
  172. file_config << "{:" << "image_name" << qrcode_images_multiple[i];
  173. std::string image_path = findDataFile(root + qrcode_images_multiple[i]);
  174. Mat src = imread(image_path);
  175. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  176. std::vector<Point> corners;
  177. QRCodeDetector qrcode;
  178. EXPECT_TRUE(qrcode.detectMulti(src, corners));
  179. std::vector<cv::String> decoded_info;
  180. std::vector<Mat> straight_barcode;
  181. EXPECT_TRUE(qrcode.decodeMulti(src, corners, decoded_info, straight_barcode));
  182. file_config << "x" << "[:";
  183. for(size_t j = 0; j < corners.size(); j += 4)
  184. {
  185. file_config << "[:";
  186. for (size_t k = 0; k < 4; k++)
  187. {
  188. file_config << corners[j + k].x;
  189. }
  190. file_config << "]";
  191. }
  192. file_config << "]";
  193. file_config << "y" << "[:";
  194. for(size_t j = 0; j < corners.size(); j += 4)
  195. {
  196. file_config << "[:";
  197. for (size_t k = 0; k < 4; k++)
  198. {
  199. file_config << corners[j + k].y;
  200. }
  201. file_config << "]";
  202. }
  203. file_config << "]";
  204. file_config << "info";
  205. file_config << "[:";
  206. for(size_t j = 0; j < decoded_info.size(); j++)
  207. {
  208. file_config << decoded_info[j];
  209. }
  210. file_config << "]";
  211. file_config << "}";
  212. }
  213. file_config << "]";
  214. file_config.release();
  215. }
  216. #else
  217. typedef testing::TestWithParam< std::string > Objdetect_QRCode;
  218. TEST_P(Objdetect_QRCode, regression)
  219. {
  220. const std::string name_current_image = GetParam();
  221. const std::string root = "qrcode/";
  222. const int pixels_error = 3;
  223. std::string image_path = findDataFile(root + name_current_image);
  224. Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
  225. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  226. std::vector<Point> corners;
  227. std::string decoded_info;
  228. QRCodeDetector qrcode;
  229. decoded_info = qrcode.detectAndDecode(src, corners, straight_barcode);
  230. ASSERT_FALSE(corners.empty());
  231. ASSERT_FALSE(decoded_info.empty());
  232. int expected_barcode_type = CV_8UC1;
  233. EXPECT_EQ(expected_barcode_type, straight_barcode.type());
  234. check_qr(root, name_current_image, "test_images", corners, {decoded_info}, pixels_error);
  235. }
  236. typedef testing::TestWithParam< std::string > Objdetect_QRCode_Close;
  237. TEST_P(Objdetect_QRCode_Close, regression)
  238. {
  239. const std::string name_current_image = GetParam();
  240. const std::string root = "qrcode/close/";
  241. const int pixels_error = 3;
  242. std::string image_path = findDataFile(root + name_current_image);
  243. Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode;
  244. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  245. const double min_side = std::min(src.size().width, src.size().height);
  246. double coeff_expansion = 1024.0 / min_side;
  247. const int width = cvRound(src.size().width * coeff_expansion);
  248. const int height = cvRound(src.size().height * coeff_expansion);
  249. Size new_size(width, height);
  250. resize(src, barcode, new_size, 0, 0, INTER_LINEAR_EXACT);
  251. std::vector<Point> corners;
  252. std::string decoded_info;
  253. QRCodeDetector qrcode;
  254. decoded_info = qrcode.detectAndDecode(barcode, corners, straight_barcode);
  255. ASSERT_FALSE(corners.empty());
  256. ASSERT_FALSE(decoded_info.empty());
  257. int expected_barcode_type = CV_8UC1;
  258. EXPECT_EQ(expected_barcode_type, straight_barcode.type());
  259. check_qr(root, name_current_image, "close_images", corners, {decoded_info}, pixels_error);
  260. }
  261. typedef testing::TestWithParam< std::string > Objdetect_QRCode_Monitor;
  262. TEST_P(Objdetect_QRCode_Monitor, regression)
  263. {
  264. const std::string name_current_image = GetParam();
  265. const std::string root = "qrcode/monitor/";
  266. const int pixels_error = 3;
  267. std::string image_path = findDataFile(root + name_current_image);
  268. Mat src = imread(image_path, IMREAD_GRAYSCALE), barcode, straight_barcode;
  269. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  270. const double min_side = std::min(src.size().width, src.size().height);
  271. double coeff_expansion = 1024.0 / min_side;
  272. const int width = cvRound(src.size().width * coeff_expansion);
  273. const int height = cvRound(src.size().height * coeff_expansion);
  274. Size new_size(width, height);
  275. resize(src, barcode, new_size, 0, 0, INTER_LINEAR_EXACT);
  276. std::vector<Point> corners;
  277. std::string decoded_info;
  278. QRCodeDetector qrcode;
  279. decoded_info = qrcode.detectAndDecode(barcode, corners, straight_barcode);
  280. ASSERT_FALSE(corners.empty());
  281. ASSERT_FALSE(decoded_info.empty());
  282. int expected_barcode_type = CV_8UC1;
  283. EXPECT_EQ(expected_barcode_type, straight_barcode.type());
  284. check_qr(root, name_current_image, "monitor_images", corners, {decoded_info}, pixels_error);
  285. }
  286. typedef testing::TestWithParam< std::string > Objdetect_QRCode_Curved;
  287. TEST_P(Objdetect_QRCode_Curved, regression)
  288. {
  289. const std::string name_current_image = GetParam();
  290. const std::string root = "qrcode/curved/";
  291. const int pixels_error = 3;
  292. std::string image_path = findDataFile(root + name_current_image);
  293. Mat src = imread(image_path, IMREAD_GRAYSCALE), straight_barcode;
  294. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  295. std::vector<Point> corners;
  296. std::string decoded_info;
  297. QRCodeDetector qrcode;
  298. decoded_info = qrcode.detectAndDecodeCurved(src, corners, straight_barcode);
  299. ASSERT_FALSE(corners.empty());
  300. ASSERT_FALSE(decoded_info.empty());
  301. int expected_barcode_type = CV_8UC1;
  302. EXPECT_EQ(expected_barcode_type, straight_barcode.type());
  303. check_qr(root, name_current_image, "test_images", corners, {decoded_info}, pixels_error);
  304. }
  305. typedef testing::TestWithParam<std::tuple<std::string, std::string>> Objdetect_QRCode_Multi;
  306. TEST_P(Objdetect_QRCode_Multi, regression)
  307. {
  308. const std::string name_current_image = get<0>(GetParam());
  309. const std::string root = "qrcode/multiple/";
  310. const std::string method = get<1>(GetParam());
  311. const int pixels_error = 4;
  312. std::string image_path = findDataFile(root + name_current_image);
  313. Mat src = imread(image_path);
  314. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  315. if (disabled_samples.find({name_current_image, method}) != disabled_samples.end())
  316. throw SkipTestException(name_current_image + " is disabled sample for method " + method);
  317. GraphicalCodeDetector qrcode = QRCodeDetector();
  318. if (method == "aruco_based") {
  319. qrcode = QRCodeDetectorAruco();
  320. }
  321. std::vector<Point> corners;
  322. std::vector<cv::String> decoded_info;
  323. std::vector<Mat> straight_barcode;
  324. EXPECT_TRUE(qrcode.detectAndDecodeMulti(src, decoded_info, corners, straight_barcode));
  325. ASSERT_FALSE(corners.empty());
  326. ASSERT_FALSE(decoded_info.empty());
  327. int expected_barcode_type = CV_8UC1;
  328. for(size_t i = 0; i < straight_barcode.size(); i++)
  329. EXPECT_EQ(expected_barcode_type, straight_barcode[i].type());
  330. check_qr(root, name_current_image, "multiple_images", corners, decoded_info, pixels_error, true);
  331. }
  332. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode, testing::ValuesIn(qrcode_images_name));
  333. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Close, testing::ValuesIn(qrcode_images_close));
  334. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Monitor, testing::ValuesIn(qrcode_images_monitor));
  335. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Curved, testing::ValuesIn(qrcode_images_curved));
  336. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_Multi, testing::Combine(testing::ValuesIn(qrcode_images_multiple),
  337. testing::Values("contours_based", "aruco_based")));
  338. TEST(Objdetect_QRCode_decodeMulti, decode_regression_16491)
  339. {
  340. Mat zero_image = Mat::zeros(256, 256, CV_8UC1);
  341. Point corners_[] = {Point(16, 16), Point(128, 16), Point(128, 128), Point(16, 128),
  342. Point(16, 16), Point(128, 16), Point(128, 128), Point(16, 128)};
  343. std::vector<Point> vec_corners;
  344. int array_size = 8;
  345. vec_corners.assign(corners_, corners_ + array_size);
  346. std::vector<cv::String> decoded_info;
  347. std::vector<Mat> straight_barcode;
  348. QRCodeDetector vec_qrcode;
  349. EXPECT_NO_THROW(vec_qrcode.decodeMulti(zero_image, vec_corners, decoded_info, straight_barcode));
  350. Mat mat_corners(2, 4, CV_32SC2, (void*)&vec_corners[0]);
  351. QRCodeDetector mat_qrcode;
  352. EXPECT_NO_THROW(mat_qrcode.decodeMulti(zero_image, mat_corners, decoded_info, straight_barcode));
  353. }
  354. typedef testing::TestWithParam<std::string> Objdetect_QRCode_detectMulti;
  355. TEST_P(Objdetect_QRCode_detectMulti, detect_regression_16961)
  356. {
  357. const std::string method = GetParam();
  358. const std::string name_current_image = "9_qrcodes.jpg";
  359. const std::string root = "qrcode/multiple/";
  360. std::string image_path = findDataFile(root + name_current_image);
  361. Mat src = imread(image_path);
  362. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  363. GraphicalCodeDetector qrcode = QRCodeDetector();
  364. if (method == "aruco_based") {
  365. qrcode = QRCodeDetectorAruco();
  366. }
  367. std::vector<Point> corners;
  368. EXPECT_TRUE(qrcode.detectMulti(src, corners));
  369. ASSERT_FALSE(corners.empty());
  370. size_t expect_corners_size = 36;
  371. EXPECT_EQ(corners.size(), expect_corners_size);
  372. }
  373. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_detectMulti, testing::Values("contours_based", "aruco_based"));
  374. typedef testing::TestWithParam<std::string> Objdetect_QRCode_detectAndDecodeMulti;
  375. TEST_P(Objdetect_QRCode_detectAndDecodeMulti, check_output_parameters_type_19363)
  376. {
  377. const std::string name_current_image = "9_qrcodes.jpg";
  378. const std::string root = "qrcode/multiple/";
  379. const std::string method = GetParam();
  380. std::string image_path = findDataFile(root + name_current_image);
  381. Mat src = imread(image_path);
  382. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  383. GraphicalCodeDetector qrcode = QRCodeDetector();
  384. if (method == "aruco_based") {
  385. qrcode = QRCodeDetectorAruco();
  386. }
  387. std::vector<Point> corners;
  388. std::vector<cv::String> decoded_info;
  389. #if 0 // FIXIT: OutputArray::create() type check
  390. std::vector<Mat2b> straight_barcode_nchannels;
  391. EXPECT_ANY_THROW(qrcode->detectAndDecodeMulti(src, decoded_info, corners, straight_barcode_nchannels));
  392. #endif
  393. int expected_barcode_type = CV_8UC1;
  394. std::vector<Mat1b> straight_barcode;
  395. EXPECT_TRUE(qrcode.detectAndDecodeMulti(src, decoded_info, corners, straight_barcode));
  396. ASSERT_FALSE(corners.empty());
  397. for(size_t i = 0; i < straight_barcode.size(); i++)
  398. EXPECT_EQ(expected_barcode_type, straight_barcode[i].type());
  399. }
  400. INSTANTIATE_TEST_CASE_P(/**/, Objdetect_QRCode_detectAndDecodeMulti, testing::Values("contours_based", "aruco_based"));
  401. TEST(Objdetect_QRCode_detect, detect_regression_20882)
  402. {
  403. const std::string name_current_image = "qrcode_near_the_end.jpg";
  404. const std::string root = "qrcode/";
  405. std::string image_path = findDataFile(root + name_current_image);
  406. Mat src = imread(image_path);
  407. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  408. QRCodeDetector qrcode;
  409. std::vector<Point> corners;
  410. Mat straight_barcode;
  411. cv::String decoded_info;
  412. EXPECT_TRUE(qrcode.detect(src, corners));
  413. EXPECT_TRUE(!corners.empty());
  414. EXPECT_NO_THROW(qrcode.decode(src, corners, straight_barcode));
  415. }
  416. TEST(Objdetect_QRCode_basic, not_found_qrcode)
  417. {
  418. std::vector<Point> corners;
  419. Mat straight_barcode;
  420. std::string decoded_info;
  421. Mat zero_image = Mat::zeros(256, 256, CV_8UC1);
  422. QRCodeDetector qrcode;
  423. EXPECT_FALSE(qrcode.detect(zero_image, corners));
  424. corners = std::vector<Point>(4);
  425. EXPECT_ANY_THROW(qrcode.decode(zero_image, corners, straight_barcode));
  426. }
  427. TEST(Objdetect_QRCode_detect, detect_regression_21287)
  428. {
  429. const std::string name_current_image = "issue_21287.png";
  430. const std::string root = "qrcode/";
  431. std::string image_path = findDataFile(root + name_current_image);
  432. Mat src = imread(image_path);
  433. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  434. QRCodeDetector qrcode;
  435. std::vector<Point> corners;
  436. Mat straight_barcode;
  437. cv::String decoded_info;
  438. EXPECT_TRUE(qrcode.detect(src, corners));
  439. EXPECT_TRUE(!corners.empty());
  440. EXPECT_NO_THROW(qrcode.decode(src, corners, straight_barcode));
  441. }
  442. TEST(Objdetect_QRCode_detect_flipped, regression_23249)
  443. {
  444. const std::vector<std::pair<std::string, std::string>> flipped_images =
  445. // image name , expected result
  446. {{"flipped_1.png", "The key is /qrcod_OMevpf"},
  447. {"flipped_2.png", "A26"}};
  448. const std::string root = "qrcode/flipped/";
  449. for(const auto &flipped_image : flipped_images){
  450. const std::string &image_name = flipped_image.first;
  451. std::string image_path = findDataFile(root + image_name);
  452. Mat src = imread(image_path);
  453. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  454. QRCodeDetector qrcode;
  455. std::vector<Point> corners;
  456. Mat straight_barcode;
  457. cv::String decoded_info;
  458. EXPECT_TRUE(qrcode.detect(src, corners));
  459. EXPECT_TRUE(!corners.empty());
  460. std::string decoded_msg;
  461. const std::string &expect_msg = flipped_image.second;
  462. EXPECT_NO_THROW(decoded_msg = qrcode.decode(src, corners, straight_barcode));
  463. ASSERT_FALSE(straight_barcode.empty()) << "Can't decode qrimage.";
  464. EXPECT_EQ(expect_msg, decoded_msg);
  465. }
  466. }
  467. // @author Kumataro, https://github.com/Kumataro
  468. TEST(Objdetect_QRCode_decode, decode_regression_21929)
  469. {
  470. const cv::String expect_msg = "OpenCV";
  471. Mat qrImg;
  472. QRCodeEncoder::Params params;
  473. params.version = 8; // 49x49
  474. Ptr<QRCodeEncoder> qrcode_enc = cv::QRCodeEncoder::create(params);
  475. qrcode_enc->encode(expect_msg, qrImg);
  476. Mat src;
  477. cv::resize(qrImg, src, Size(200,200), 1.0, 1.0, INTER_NEAREST);
  478. QRCodeDetector qrcode;
  479. std::vector<Point> corners;
  480. Mat straight_barcode;
  481. EXPECT_TRUE(qrcode.detect(src, corners));
  482. EXPECT_TRUE(!corners.empty());
  483. cv::String decoded_msg;
  484. EXPECT_NO_THROW(decoded_msg = qrcode.decode(src, corners, straight_barcode));
  485. ASSERT_FALSE(straight_barcode.empty()) << "Can't decode qrimage.";
  486. EXPECT_EQ(expect_msg, decoded_msg);
  487. }
  488. TEST(Objdetect_QRCode_decode, decode_regression_version_25)
  489. {
  490. const cv::String expect_msg = "OpenCV";
  491. Mat qrImg;
  492. QRCodeEncoder::Params params;
  493. params.version = 25; // 117x117
  494. Ptr<QRCodeEncoder> qrcode_enc = cv::QRCodeEncoder::create(params);
  495. qrcode_enc->encode(expect_msg, qrImg);
  496. Mat src;
  497. cv::resize(qrImg, src, qrImg.size()*3, 1.0, 1.0, INTER_NEAREST);
  498. QRCodeDetector qrcode;
  499. std::vector<Point> corners;
  500. Mat straight_barcode;
  501. EXPECT_TRUE(qrcode.detect(src, corners));
  502. EXPECT_TRUE(!corners.empty());
  503. cv::String decoded_msg;
  504. EXPECT_NO_THROW(decoded_msg = qrcode.decode(src, corners, straight_barcode));
  505. ASSERT_FALSE(straight_barcode.empty()) << "Can't decode qrimage.";
  506. EXPECT_EQ(expect_msg, decoded_msg);
  507. }
  508. TEST_P(Objdetect_QRCode_detectAndDecodeMulti, decode_9_qrcodes_version7)
  509. {
  510. const std::string name_current_image = "9_qrcodes_version7.jpg";
  511. const std::string root = "qrcode/multiple/";
  512. std::string image_path = findDataFile(root + name_current_image);
  513. Mat src = imread(image_path);
  514. const std::string method = GetParam();
  515. GraphicalCodeDetector qrcode = QRCodeDetector();
  516. if (method == "aruco_based") {
  517. qrcode = QRCodeDetectorAruco();
  518. }
  519. std::vector<Point> corners;
  520. std::vector<cv::String> decoded_info;
  521. std::vector<Mat1b> straight_barcode;
  522. qrcode.detectAndDecodeMulti(src, decoded_info, corners, straight_barcode);
  523. EXPECT_EQ(9ull, decoded_info.size());
  524. const string gold_info = "I love OpenCV, QR Code version = 7, error correction = level Quartile";
  525. for (const auto& info : decoded_info) {
  526. EXPECT_EQ(info, gold_info);
  527. }
  528. }
  529. #endif // UPDATE_QRCODE_TEST_DATA
  530. TEST(Objdetect_QRCode_detectAndDecode, utf8_output)
  531. {
  532. const std::string name_current_image = "umlaut.png";
  533. const std::string root = "qrcode/";
  534. std::string image_path = findDataFile(root + name_current_image);
  535. Mat src = imread(image_path);
  536. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  537. QRCodeDetector qrcode;
  538. std::vector<Point> corners;
  539. Mat straight;
  540. std::string decoded_info = qrcode.detectAndDecode(src, corners, straight);
  541. EXPECT_FALSE(decoded_info.empty());
  542. EXPECT_NE(decoded_info.find("M\xc3\xbcllheimstrasse"), std::string::npos);
  543. }
  544. TEST_P(Objdetect_QRCode_detectAndDecodeMulti, detect_regression_24679)
  545. {
  546. const std::string name_current_image = "issue_24679.png";
  547. const std::string root = "qrcode/";
  548. std::string image_path = findDataFile(root + name_current_image);
  549. Mat img = imread(image_path);
  550. const std::string method = GetParam();
  551. GraphicalCodeDetector qrcode = QRCodeDetector();
  552. if (method == "aruco_based") {
  553. qrcode = QRCodeDetectorAruco();
  554. }
  555. std::vector<cv::String> decoded_info;
  556. ASSERT_TRUE(qrcode.detectAndDecodeMulti(img, decoded_info));
  557. EXPECT_EQ(decoded_info.size(), 4U);
  558. }
  559. TEST_P(Objdetect_QRCode_detectAndDecodeMulti, detect_regression_24011)
  560. {
  561. const std::string name_current_image = "issue_24011.jpg";
  562. const std::string root = "qrcode/";
  563. std::string image_path = findDataFile(root + name_current_image);
  564. Mat img = imread(image_path);
  565. const std::string method = GetParam();
  566. GraphicalCodeDetector qrcode = QRCodeDetector();
  567. if (method == "aruco_based") {
  568. qrcode = QRCodeDetectorAruco();
  569. }
  570. std::vector<cv::String> decoded_info;
  571. ASSERT_TRUE(qrcode.detectAndDecodeMulti(img, decoded_info));
  572. EXPECT_EQ(decoded_info.size(), 2U);
  573. }
  574. TEST(Objdetect_QRCode_detect, detect_regression_24450)
  575. {
  576. const std::string name_current_image = "issue_24450.png";
  577. const std::string root = "qrcode/";
  578. std::string image_path = findDataFile(root + name_current_image);
  579. Mat img = imread(image_path);
  580. GraphicalCodeDetector qrcode = QRCodeDetector();
  581. std::vector<Point2f> points;
  582. ASSERT_TRUE(qrcode.detect(img, points));
  583. EXPECT_EQ(points.size(), 4U);
  584. img.at<Vec3b>(img.rows - 1, 296) = {};
  585. ASSERT_TRUE(qrcode.detect(img, points));
  586. EXPECT_EQ(points.size(), 4U);
  587. }
  588. TEST(Objdetect_QRCode_detect, detect_regression_22892)
  589. {
  590. const std::string name_current_image = "issue_22892.png";
  591. const std::string root = "qrcode/";
  592. std::string image_path = findDataFile(root + name_current_image);
  593. Mat img = imread(image_path);
  594. QRCodeDetector qrcode;
  595. std::vector<Point> corners;
  596. Mat straight_code;
  597. qrcode.detectAndDecodeCurved(img, corners, straight_code);
  598. EXPECT_EQ(corners.size(), 4U);
  599. }
  600. // See https://github.com/opencv/opencv/issues/27783
  601. TEST(Objdetect_QRCode_detect, detect_regression_27783)
  602. {
  603. const std::string name_current_image = "9_qrcodes.jpg";
  604. const std::string root = "qrcode/multiple/";
  605. std::string image_path = findDataFile(root + name_current_image);
  606. Mat src = imread(image_path);
  607. ASSERT_FALSE(src.empty()) << "Can't read image: " << image_path;
  608. std::vector<std::string> info;
  609. std::vector<Point> corners;
  610. {
  611. // If default, we can decode 9 QRs.
  612. QRCodeDetector qrcode;
  613. EXPECT_TRUE(qrcode.detectAndDecodeMulti(src, info, corners));
  614. ASSERT_EQ(info.size(), 9UL);
  615. ASSERT_EQ(corners.size(), 36UL);
  616. }
  617. {
  618. // If setEpsX is too small, we can decode no QRs.
  619. QRCodeDetector qrcode;
  620. qrcode.setEpsX(0.01);
  621. EXPECT_FALSE(qrcode.detectAndDecodeMulti(src, info, corners));
  622. }
  623. {
  624. // If setEpsY is too small, we can decode no QRs.
  625. QRCodeDetector qrcode;
  626. qrcode.setEpsY(0.01);
  627. EXPECT_FALSE(qrcode.detectAndDecodeMulti(src, info, corners));
  628. }
  629. }
  630. }} // namespace