Python OpenCV 去除截图中的所有图片

import cv2
import numpy as np
import matplotlib.pyplot as plt

# 读取图像并转换为灰度图像
image = cv2.imread('./a.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 使用 Canny 边缘检测
edges = cv2.Canny(gray, 0, 200)  # 调整阈值参数

# 寻找轮廓
contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# 过滤出长宽大于150的轮廓
filtered_contours = [contour for contour in contours if cv2.boundingRect(contour)[2] > 110 and cv2.boundingRect(contour)[3] > 110]

# 对轮廓按面积进行排序
sorted_contours = sorted(filtered_contours, key=cv2.contourArea, reverse=True)[:10]  # 只保留最大的五个轮廓

# 绘制矩形轮廓
for contour in sorted_contours:
    x, y, w, h = cv2.boundingRect(contour)
    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)
    
# 创建一个与原始图像相同大小的空白图像
mask = np.zeros_like(image)

# 将轮廓内的区域更改为背景颜色
for contour in sorted_contours:
    x, y, w, h = cv2.boundingRect(contour)
    cv2.rectangle(mask, (x, y), (x + w, y + h), (255, 255, 255), thickness=cv2.FILLED)


# 将填充区域与原始图像进行叠加
result = np.where(mask != 255, image, mask)
# result = cv2.bitwise_and(image, cv2.bitwise_not(mask))
# 显示结果
plt.imshow(result[:,:,::-1])