Введение
Эта статья — первая в цикле статей, в котором мы разберемся с тем, как qr-код устроен, и напишем простенький Qr-детектор и дешифровщик, а также свой собственный генератор qr-кодов.
Использовать мы будем python вместе с opencv и numpy. Учитывая, что opencv — кросс-язычная библиотека, а также то, что работа с изображением/текстурой в разных решениях выглядят примерно одинаково, то я думаю, что вы без труда сможете перевести алгоритм, который будет здесь написан, на любой нужный вам язык
В первую очередь мы будем рассматривать полноразмерный qr-код, Micro-qr возможно будет рассмотрен после завершения работы над полноразмерным qr
Также, хочу отметить, что готовый класс QrCodeDetector уже имеется внутри opencv. Возможно, вам не нужно изобретать велосипед 🙂
Обнаружение
Очевидно, что прежде, чем дешифровать qr-код, нужно для начала его обнаружить на картинке. Как же это делают наши смартфоны? Всё очень просто, специально для этого на Qr-коде есть вот эти три квадратика:
Пишем код
Как уже было сказано выше, использовать мы будем opencv и numpy. Импортируем эти библиотеки:
import cv2 as cv import numpy as npВ первую очередь нам нужно найти первый чёрный пиксель на изображении, которое является трёхмерным массивом вида:
img[y, x, 1]Поэтому мы проходимся по массиву, пока не найдем элемент, значение которого меньше 50. (черный цвет = 0, но на изображении могут быть помехи, так что мы просто ищем тёмные пиксели):
class QrHandler(): def detect(self, img): for y in range(0, len(img)): for x in range(0, len(img[0])): if (img[y, x] < [50, 50, 50]).all(): print('black')Кстати, ради дебага я использую режим чтения cv.IMREAD_COLOR, по существу он здесь совершенно не нужен, так что я советую заменить его на cv.IMREAD_GRAYSCALE
Мы нашли черный пиксель, теперь нам нужно проверить, что он является частью квадрата:
import cv2 as cv import numpy as np class QrHandler(): def detect(self, img): for y in range(0, len(img)): for x in range(0, len(img[0])): if (img[y, x] < [50, 50, 50]).all(): square_length = self._get_square_length(img, y, x) if square_length != -5: prtint('square') # не забываем про помехи на изображении, поэтому при проверке какой-либо точки нужно проверять небольшой регион вокруг этой точки def _is_black_point(self, img, y, x, inaccuracy): y_2 = y + inaccuracy if y_2 >= len(img): y_2 = len(img) - 1 x_2 = x + inaccuracy if x_2 >= len(img[0]): x_2 = len(img[0]) - 1 for y in range(y - inaccuracy, y_2): for x in range(x - inaccuracy, x_2): if (img[y, x] < [50, 50, 50]).all(): return True return False def _get_square_length(self, img, y, x): square_length = 0 # идём вправо по x и ищем конец квадрата, находим его примерную длину for x_i in range(x, len(img[0])): if (img[y, x_i] > [50, 50, 50]).all(): break square_length += 1 # слишком маленькая длина явно говорит нам о том, что это неподходящий квадратик, поэтому проверяем if square_length >= 6: #проверяем две точки: по y и по диагонали if self._is_black_point(img, y + square_length, x + square_length, 3) and self._is_black_point(img, y + square_length, x, 3): return square_length return -5Здесь в функции get_square_length мы сначала ищем длину квадрата, проходя по нему до сюда:
А затем при помощи функции _is_black_point проверяем два региона:
Мы проверяем именно два региона, а не две точки, т.к., как уже было сказано ранее, существует погрешность при работе с изображением
Также, стоит проверить наличие вот этого маленького квадратика внутри:
Для этого в наш класс пишем еще одну функцию:
class QrHandler(): def detect(self, img): for y in range(0, len(img)): for x in range(0, len(img[0])): if (img[y, x] < [50, 50, 50]).all(): square_length = self._get_square_length(img, y, x) #добавляем нашу новую функцию в проверку квадратика if square_length != -5 and self._is_has_lil_square(img, y, x, square_length): print('square') def _is_has_lil_square(self, img, y, x, square_length): lil_square_length = 0 #находим центр найденного нами квадрата y = y + square_length // 2 x = x + square_length // 2 have_white = False #идем от центра, пока не найдем границу квадратика #запоминаем расстояние от центра до границы for x_lil in range(x, x + square_length): if (img[y, x_lil] > [50, 50, 50]).all(): have_white = True break lil_square_length += 1 if have_white: have_white = False lil_square_length_y = 0 #если мы нашли границу по x, то потвторяем то же самое по y for y_lil in range(y, y + square_length): if (img[y_lil, x] > [50, 50, 50]).all(): have_white = True break lil_square_length_y += 1 #если нашли границу по y, то нужно проверить расстояние до нее от центра #расстояние по x и по y должно быть примерно равно друг другу if have_white and (lil_square_length_y in range(lil_square_length - 3, lil_square_length + 3)): #также нужно проверить нижнюю правую точку квадратика if self._is_black_point(img, y + lil_square_length, x + lil_square_length, 3): return True return FalseТеперь похожим образом нужно проверить оставшиеся два квадрата, для этого дописываем нашу detect функцию:
class QrHandler(): def detect(self, img): for y in range(0, len(img)): for x in range(0, len(img[0])): if (img[y, x] < [50, 50, 50]).all(): square_length = self._get_square_length(img, y, x) if square_length != -5 and self._is_has_lil_square(img, y, x, square_length): #перебираем точки по y for y_2 in range(y + square_length, len(img)): if (img[y_2, x] < [50, 50, 50]).all(): square_length_2 = self._get_square_length( img, y_2, x) if square_length_2 != -5 and self._is_has_lil_square(img, y_2, x, square_length_2): #после того как нашли потенциальный квадрат, #нужно проверить, что его длина примерно равна длине уже найденного квадрата if square_length_2 in range(square_length - 3, square_length + 3): qr_size = y_2 - y #мы уже знаем расстояние между двумя квадратиками, поэтому нам не нужно проходиться по точкам #сразу проверяем потенциальную точку square_length_3 = self._get_square_length( img, y, x + qr_size) if square_length_3 != -5 and self._is_has_lil_square(img, y, x + qr_size, square_length_3): if square_length_3 in range(square_length - 3, square_length + 3): #проверяем квадратик по аналогии со вторм и возвращаем вырезанный qr return img[y: y + qr_size + square_length, x: x + qr_size + square_length]По итогу мы получаем следующий код:
import cv2 as cv import numpy as np class QrHandler(): def detect(self, img): for y in range(0, len(img)): for x in range(0, len(img[0])): if (img[y, x] < [50, 50, 50]).all(): square_length = self._get_square_length(img, y, x) if square_length != -5 and self._is_has_lil_square(img, y, x, square_length): for y_2 in range(y + square_length, len(img)): if (img[y_2, x] < [50, 50, 50]).all(): square_length_2 = self._get_square_length( img, y_2, x) if square_length_2 != -5 and self._is_has_lil_square(img, y_2, x, square_length_2): if square_length_2 in range(square_length - 3, square_length + 3): qr_size = y_2 - y square_length_3 = self._get_square_length( img, y, x + qr_size) if square_length_3 != -5 and self._is_has_lil_square(img, y, x + qr_size, square_length_3): if square_length_3 in range(square_length - 3, square_length + 3): return img[y: y + qr_size + square_length, x: x + qr_size + square_length] # не забываем про помехи на изображении, поэтому при проверке какой-либо точки нужно проверять небольшой регион вокруг этой точки def _is_black_point(self, img, y, x, inaccuracy): y_2 = y + inaccuracy if y_2 >= len(img): y_2 = len(img) - 1 x_2 = x + inaccuracy if x_2 >= len(img[0]): x_2 = len(img[0]) - 1 for y in range(y - inaccuracy, y_2): for x in range(x - inaccuracy, x_2): if (img[y, x] < [50, 50, 50]).all(): return True return False def _get_square_length(self, img, y, x): square_length = 0 # идём вправо и ищем конец квадрата, ищем его примерную длину for x_i in range(x, len(img[0])): if (img[y, x_i] > [50, 50, 50]).all(): break square_length += 1 # слишком маленькая длина явно говорит нам о том, что это неподходящий квадратик, поэтому проверяем if square_length >= 6: if self._is_black_point(img, y + square_length, x + square_length, 3) and self._is_black_point(img, y + square_length, x, 3): return square_length return -5 def _is_has_lil_square(self, img, y, x, square_length): lil_square_length = 0 y = y + square_length // 2 x = x + square_length // 2 have_white = False for x_lil in range(x, x + square_length): if (img[y, x_lil] > [50, 50, 50]).all(): have_white = True break lil_square_length += 1 if have_white: have_white = False lil_square_length_y = 0 for y_lil in range(y, y + square_length): if (img[y_lil, x] > [50, 50, 50]).all(): have_white = True break lil_square_length_y += 1 if have_white and (lil_square_length_y in range(lil_square_length - 3, lil_square_length + 3)): if self._is_black_point(img, y + lil_square_length, x + lil_square_length, 3): return True return False #первый аргумент этой функции - наименование вашего изображения в одной папке с исполняемым файлом img = cv.imread('qr_wider.jpg', cv.IMREAD_COLOR) qr_handler = QrHandler() img = qr_handler.detect(img) cv.imshow('test', img) cv.waitKey(0) Заключение
В этой статье мы написали простенький qr-детектор, который обнаруживает qr-код на белом фоне. В следующих статьях мы научимся обнаруживать qr-код на более сложных изображениях, после чего переводить его в понимаемы нами (человеками) формат
Код всех частей этого цикла можно найти в этом репозитории
Если вам необходимо получить полную информацию о qr-кодах, не ожидая выхода всех частей этого цикла, советую ознакомиться с данной документацией: ISO/IEC JTC 1/SC 31 N (arscreatio.com). Сущность qr-кода не сильно изменилась со времен его создания, поэтому не смотрите на то, что документация 2004 года
*нашел вариант на русском из нашего ГОСТ-а: ГОСТ Р ИСО/МЭК 18004-2015. Информационные технологии. Технологии автоматической идентификации и сбора данных. Спецификация символики штрихового кода QR Code (internet-law.ru)
Также, для быстрого понимания основных принципов чтения qr-кода, советую обратить внимание на эту статью.
ссылка на оригинал статьи https://habr.com/ru/post/708832/
Добавить комментарий