mirror of
https://github.com/opencv/opencv.git
synced 2024-12-13 07:59:27 +08:00
251 lines
8.3 KiB
Python
251 lines
8.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
'''
|
|
This program detects the QR-codes using OpenCV Library.
|
|
|
|
Usage:
|
|
qrcode.py
|
|
'''
|
|
|
|
|
|
# Python 2/3 compatibility
|
|
from __future__ import print_function
|
|
|
|
import numpy as np
|
|
import cv2 as cv
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
PY3 = sys.version_info[0] == 3
|
|
if PY3:
|
|
xrange = range
|
|
|
|
|
|
class QrSample:
|
|
def __init__(self, args):
|
|
self.fname = ''
|
|
self.fext = ''
|
|
self.fsaveid = 0
|
|
self.input = args.input
|
|
self.detect = args.detect
|
|
self.out = args.out
|
|
self.multi = args.multi
|
|
self.saveDetections = args.save_detections
|
|
self.saveAll = args.save_all
|
|
|
|
def getQRModeString(self):
|
|
msg1 = "multi " if self.multi else ""
|
|
msg2 = "detector" if self.detect else "decoder"
|
|
msg = "QR {:s}{:s}".format(msg1, msg2)
|
|
return msg
|
|
|
|
def drawFPS(self, result, fps):
|
|
message = '{:.2f} FPS({:s})'.format(fps, self.getQRModeString())
|
|
cv.putText(result, message, (20, 20), 1,
|
|
cv.FONT_HERSHEY_DUPLEX, (0, 0, 255))
|
|
|
|
def drawQRCodeContours(self, image, cnt):
|
|
if cnt.size != 0:
|
|
rows, cols, _ = image.shape
|
|
show_radius = 2.813 * ((rows / cols) if rows > cols else (cols / rows))
|
|
contour_radius = show_radius * 0.4
|
|
cv.drawContours(image, [cnt], 0, (0, 255, 0), int(round(contour_radius)))
|
|
tpl = cnt.reshape((-1, 2))
|
|
for x in tuple(tpl.tolist()):
|
|
color = (255, 0, 0)
|
|
cv.circle(image, tuple(x), int(round(contour_radius)), color, -1)
|
|
|
|
def drawQRCodeResults(self, result, points, decode_info, fps):
|
|
n = len(points)
|
|
if isinstance(decode_info, str):
|
|
decode_info = [decode_info]
|
|
if n > 0:
|
|
for i in range(n):
|
|
cnt = np.array(points[i]).reshape((-1, 1, 2)).astype(np.int32)
|
|
self.drawQRCodeContours(result, cnt)
|
|
msg = 'QR[{:d}]@{} : '.format(i, *(cnt.reshape(1, -1).tolist()))
|
|
print(msg, end="")
|
|
if len(decode_info) > i:
|
|
if decode_info[i]:
|
|
print("'", decode_info[i], "'")
|
|
else:
|
|
print("Can't decode QR code")
|
|
else:
|
|
print("Decode information is not available (disabled)")
|
|
else:
|
|
print("QRCode not detected!")
|
|
self.drawFPS(result, fps)
|
|
|
|
def runQR(self, qrCode, inputimg):
|
|
if not self.multi:
|
|
if not self.detect:
|
|
decode_info, points, _ = qrCode.detectAndDecode(inputimg)
|
|
dec_info = decode_info
|
|
else:
|
|
_, points = qrCode.detect(inputimg)
|
|
dec_info = []
|
|
else:
|
|
if not self.detect:
|
|
_, decode_info, points, _ = qrCode.detectAndDecodeMulti(
|
|
inputimg)
|
|
dec_info = decode_info
|
|
else:
|
|
_, points = qrCode.detectMulti(inputimg)
|
|
dec_info = []
|
|
if points is None:
|
|
points = []
|
|
return points, dec_info
|
|
|
|
def DetectQRFrmImage(self, inputfile):
|
|
inputimg = cv.imread(inputfile, cv.IMREAD_COLOR)
|
|
if inputimg is None:
|
|
print('ERROR: Can not read image: {}'.format(inputfile))
|
|
return
|
|
print('Run {:s} on image [{:d}x{:d}]'.format(
|
|
self.getQRModeString(), inputimg.shape[1], inputimg.shape[0]))
|
|
qrCode = cv.QRCodeDetector()
|
|
count = 10
|
|
timer = cv.TickMeter()
|
|
for _ in range(count):
|
|
timer.start()
|
|
points, decode_info = self.runQR(qrCode, inputimg)
|
|
timer.stop()
|
|
fps = count / timer.getTimeSec()
|
|
print('FPS: {}'.format(fps))
|
|
result = inputimg
|
|
self.drawQRCodeResults(result, points, decode_info, fps)
|
|
cv.imshow("QR", result)
|
|
cv.waitKey(1)
|
|
if self.out != '':
|
|
outfile = self.fname + self.fext
|
|
print("Saving Result: {}".format(outfile))
|
|
cv.imwrite(outfile, result)
|
|
|
|
print("Press any key to exit ...")
|
|
cv.waitKey(0)
|
|
print("Exit")
|
|
|
|
def processQRCodeDetection(self, qrcode, frame):
|
|
if len(frame.shape) == 2:
|
|
result = cv.cvtColor(frame, cv.COLOR_GRAY2BGR)
|
|
else:
|
|
result = frame
|
|
print('Run {:s} on video frame [{:d}x{:d}]'.format(
|
|
self.getQRModeString(), frame.shape[1], frame.shape[0]))
|
|
timer = cv.TickMeter()
|
|
timer.start()
|
|
points, decode_info = self.runQR(qrcode, frame)
|
|
timer.stop()
|
|
|
|
fps = 1 / timer.getTimeSec()
|
|
self.drawQRCodeResults(result, points, decode_info, fps)
|
|
return fps, result, points
|
|
|
|
def DetectQRFrmCamera(self):
|
|
cap = cv.VideoCapture(0)
|
|
if not cap.isOpened():
|
|
print("Cannot open the camera")
|
|
return
|
|
print("Press 'm' to switch between detectAndDecode and detectAndDecodeMulti")
|
|
print("Press 'd' to switch between decoder and detector")
|
|
print("Press ' ' (space) to save result into images")
|
|
print("Press 'ESC' to exit")
|
|
|
|
qrcode = cv.QRCodeDetector()
|
|
|
|
while True:
|
|
ret, frame = cap.read()
|
|
if not ret:
|
|
print("End of video stream")
|
|
break
|
|
forcesave = self.saveAll
|
|
result = frame
|
|
try:
|
|
fps, result, corners = self.processQRCodeDetection(qrcode, frame)
|
|
print('FPS: {:.2f}'.format(fps))
|
|
forcesave |= self.saveDetections and (len(corners) != 0)
|
|
except cv.error as e:
|
|
print("Error exception: ", e)
|
|
forcesave = True
|
|
cv.imshow("QR code", result)
|
|
code = cv.waitKey(1)
|
|
if code < 0 and (not forcesave):
|
|
continue
|
|
if code == ord(' ') or forcesave:
|
|
fsuffix = '-{:05d}'.format(self.fsaveid)
|
|
self.fsaveid += 1
|
|
fname_in = self.fname + fsuffix + "_input.png"
|
|
print("Saving QR code detection result: '{}' ...".format(fname_in))
|
|
cv.imwrite(fname_in, frame)
|
|
print("Saved")
|
|
if code == ord('m'):
|
|
self.multi = not self.multi
|
|
msg = 'Switching QR code mode ==> {:s}'.format(
|
|
"detectAndDecodeMulti" if self.multi else "detectAndDecode")
|
|
print(msg)
|
|
if code == ord('d'):
|
|
self.detect = not self.detect
|
|
msg = 'Switching QR code mode ==> {:s}'.format(
|
|
"detect" if self.detect else "decode")
|
|
print(msg)
|
|
if code == 27:
|
|
print("'ESC' is pressed. Exiting...")
|
|
break
|
|
print("Exit.")
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description='This program detects the QR-codes input images using OpenCV Library.')
|
|
parser.add_argument(
|
|
'-i',
|
|
'--input',
|
|
help="input image path (for example, 'opencv_extra/testdata/cv/qrcode/multiple/*_qrcodes.png)",
|
|
default="",
|
|
metavar="")
|
|
parser.add_argument(
|
|
'-d',
|
|
'--detect',
|
|
help="detect QR code only (skip decoding) (default: False)",
|
|
action='store_true')
|
|
parser.add_argument(
|
|
'-m',
|
|
'--multi',
|
|
help="enable multiple qr-codes detection",
|
|
action='store_true')
|
|
parser.add_argument(
|
|
'-o',
|
|
'--out',
|
|
help="path to result file (default: qr_code.png)",
|
|
default="qr_code.png",
|
|
metavar="")
|
|
parser.add_argument(
|
|
'--save_detections',
|
|
help="save all QR detections (video mode only)",
|
|
action='store_true')
|
|
parser.add_argument(
|
|
'--save_all',
|
|
help="save all processed frames (video mode only)",
|
|
action='store_true')
|
|
args = parser.parse_args()
|
|
qrinst = QrSample(args)
|
|
if args.out != '':
|
|
index = args.out.rfind('.')
|
|
if index != -1:
|
|
qrinst.fname = args.out[:index]
|
|
qrinst.fext = args.out[index:]
|
|
else:
|
|
qrinst.fname = args.out
|
|
qrinst.fext = ".png"
|
|
if args.input != '':
|
|
qrinst.DetectQRFrmImage(args.input)
|
|
else:
|
|
qrinst.DetectQRFrmCamera()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print(__doc__)
|
|
main()
|
|
cv.destroyAllWindows()
|