Fix bbox coordinates extraction

AlexBulavin · web-flow · commit 9110926cc4c2 · 2022-01-15T23:34:48.000+03:00
All commented in code, please see there
diff --git a/general/generating-reading-qrcode/read_qrcode.py b/general/generating-reading-qrcode/read_qrcode.py
@@ -4,7 +4,9 @@
 filename = sys.argv[1]
 
 # read the QRCODE image
-img = cv2.imread(filename)
+#in case if QR code is not black/white it is better to convert it into grayscale
+img = cv2.imread(filename, 0)# Zero means grayscale
+img_origin = cv2.imread(filename)
 
 # initialize the cv2 QRCode detector
 detector = cv2.QRCodeDetector()
@@ -17,16 +19,19 @@
     print(f"QRCode data:\n{data}")
     # display the image with lines
     # length of bounding box
-    n_lines = len(bbox)
+    n_lines = len(bbox[0])#Cause bbox = [[[float, float]]], we need to convert fload into int and loop over the first element of array
+    bbox1 = bbox.astype(int) #Float to Int conversion
     for i in range(n_lines):
         # draw all lines
-        point1 = tuple(bbox[i][0])
-        point2 = tuple(bbox[(i+1) % n_lines][0])
-        cv2.line(img, point1, point2, color=(255, 0, 0), thickness=2)
-
-
-
-# display the result
-cv2.imshow("img", img)
-cv2.waitKey(0)
-cv2.destroyAllWindows()
+        point1 = tuple(bbox1[0, [i][0]])
+        point2 = tuple(bbox1[0, [(i+1) % n_lines][0]])
+        cv2.line(img_origin, point1, point2, color=(255, 0, 0), thickness=2)
+
+
+    # display the result
+    cv2.imshow("img", img)
+    cv2.waitKey(0)
+    cv2.destroyAllWindows()
+    
+else:
+    print("QR code not detected")
