# -*- coding: utf-8 -*-
# @Time    : 2024/2/18 14:23
# @Author  : len
# @File    : find_qr_block.py
# @Software: PyCharm 
# @Comment :

import cv2
import numpy as np

# 读取图像
image_path = './android_img/img_240218/img_5.png'  # 替换为你的图像文件路径
image = cv2.imread(image_path)

# 灰度化
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 应用阈值处理或者Canny边缘检测
# ret, thresh = cv2.threshold(gray, 127, 255, 0)  # 二值化阈值处理
# thresh = cv2.Canny(image, 50, 150)  # 或者使用Canny边缘检测
thresh = cv2.Canny(gray, 50, 150)  # 或者使用Canny边缘检测

# 寻找轮廓
contours, _ = cv2.findContours(thresh, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)

# 筛选和绘制轮廓
for cnt in contours:
    # 轮廓近似
    epsilon = 0.05 * cv2.arcLength(cnt, True)
    approx = cv2.approxPolyDP(cnt, epsilon, True)

    # 如果近似轮廓有4个顶点，则可能是矩形或方块
    if len(approx) == 4:
        # 绘制轮廓
        cv2.drawContours(image, [approx], -1, (0, 255, 0), 3)

# 显示图像
cv2.imshow('gray', gray)
cv2.imshow('Image with Blocks', image)
cv2.waitKey(0)
cv2.destroyAllWindows()