You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.7 KiB
53 lines
1.7 KiB
import cv2 |
|
import sys |
|
|
|
print(sys.argv[0]) |
|
print('A demo program of WeChat QRCode Detector:') |
|
camIdx = -1 |
|
if len(sys.argv) > 1: |
|
if sys.argv[1] == "-camera": |
|
camIdx = int(sys.argv[2]) if len(sys.argv)>2 else 0 |
|
img = cv2.imread(sys.argv[1]) |
|
else: |
|
print(" Usage: " + sys.argv[0] + " <input_image>") |
|
exit(0) |
|
|
|
# For python API generator, it follows the template: {module_name}_{class_name}, |
|
# so it is a little weird. |
|
# The model is downloaded to ${CMAKE_BINARY_DIR}/downloads/wechat_qrcode if cmake runs without warnings, |
|
# otherwise you can download them from https://github.com/WeChatCV/opencv_3rdparty/tree/wechat_qrcode. |
|
try: |
|
detector = cv2.wechat_qrcode_WeChatQRCode( |
|
"detect.prototxt", "detect.caffemodel", "sr.prototxt", "sr.caffemodel") |
|
except: |
|
print("---------------------------------------------------------------") |
|
print("Failed to initialize WeChatQRCode.") |
|
print("Please, download 'detector.*' and 'sr.*' from") |
|
print("https://github.com/WeChatCV/opencv_3rdparty/tree/wechat_qrcode") |
|
print("and put them into the current directory.") |
|
print("---------------------------------------------------------------") |
|
exit(0) |
|
|
|
prevstr = "" |
|
|
|
if camIdx < 0: |
|
res, points = detector.detectAndDecode(img) |
|
print(res,points) |
|
else: |
|
cap = cv2.VideoCapture(camIdx) |
|
while True: |
|
res, img = cap.read() |
|
if img is None: |
|
break |
|
res, points = detector.detectAndDecode(img) |
|
for t in res: |
|
if t != prevstr: |
|
print(t) |
|
if res: |
|
prevstr = res[-1] |
|
cv2.imshow("image", img) |
|
if cv2.waitKey(30) >= 0: |
|
break |
|
# When everything done, release the capture |
|
cap.release() |
|
cv2.destroyAllWindows()
|
|
|