mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import logging
|
|
import os
|
|
from datetime import datetime
|
|
|
|
import cv2
|
|
|
|
from fishy.helper.helper import get_documents
|
|
|
|
import numpy as np
|
|
from pyzbar.pyzbar import decode
|
|
|
|
|
|
def get_qr_location(og_img):
|
|
"""
|
|
code from https://stackoverflow.com/a/45770227/4512396
|
|
"""
|
|
gray = cv2.bilateralFilter(og_img, 11, 17, 17)
|
|
kernel = np.ones((5, 5), np.uint8)
|
|
erosion = cv2.erode(gray, kernel, iterations=2)
|
|
kernel = np.ones((4, 4), np.uint8)
|
|
img = cv2.dilate(erosion, kernel, iterations=2)
|
|
|
|
cnt, h = cv2.findContours(img, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
|
|
|
|
valid_crops = []
|
|
for i in range(len(cnt)):
|
|
area = cv2.contourArea(cnt[i])
|
|
if 500 < area < 100000:
|
|
mask = np.zeros_like(img)
|
|
cv2.drawContours(mask, cnt, i, 255, -1)
|
|
x, y, w, h = cv2.boundingRect(cnt[i])
|
|
qr_result = decode(og_img[y:h + y, x:w + x])
|
|
if qr_result:
|
|
valid_crops.append(((x, y, x + w, y + h), area))
|
|
|
|
return min(valid_crops, key=lambda c: c[1])[0] if valid_crops else None
|
|
|
|
|
|
# noinspection PyBroadException
|
|
def get_values_from_image(img):
|
|
try:
|
|
for qr in decode(img):
|
|
vals = qr.data.decode('utf-8').split(",")
|
|
return float(vals[0]), float(vals[1]), float(vals[2])
|
|
|
|
logging.error("FishyQR not found, try restarting the engine")
|
|
return None
|
|
except Exception:
|
|
logging.error("Couldn't read coods, make sure 'crop' calibration is correct")
|
|
cv2.imwrite(os.path.join(get_documents(), "fishy_failed_reads", f"{datetime.now()}.jpg"), img)
|
|
return None
|