使用Python移除带有水印的图片上的黑色边框

2024-04-25 04:41:50 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一堆图像,我想通过删除黑色边框来统一。通常我使用带有模糊参数的Imagemagick的Trim函数,但是如果图像有水印,结果就不在这里。

实际上,我正在用opencv和形态变换做一些测试,试图识别水印和图像,然后选择更大的元素,但我真的是新的opencv和我挣扎。

水印可以无处不在,从左下到右上。

black borders with watermak 1black borders with watermak 2black borders with watermak 3

我更喜欢Python代码,但欢迎使用Imagemagick或类似的应用程序。

实际上,只有使用opencv才能得到这样的结果:

import copy
import cv2

from matplotlib import pyplot as plt


IMG_IN = '/data/black_borders/island.jpg'


# keep a copy of original image
original = cv2.imread(IMG_IN)

# Read the image, convert it into grayscale, and make in binary image for threshold value of 1.
img = cv2.imread(IMG_IN,0)

# use binary threshold, all pixel that are beyond 3 are made white
_, thresh_original = cv2.threshold(img, 3, 255, cv2.THRESH_BINARY)

# Now find contours in it.
thresh = copy.copy(thresh_original)
contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

# get contours with highest height
lst_contours = []
for cnt in contours:
    ctr = cv2.boundingRect(cnt)
    lst_contours.append(ctr)
x,y,w,h = sorted(lst_contours, key=lambda coef: coef[3])[-1]


# draw contours
ctr = copy.copy(original)
cv2.rectangle(ctr, (x,y),(x+w,y+h),(0,255,0),2)


# display results with matplotlib

# original
original = original[:,:,::-1] # flip color for maptolib display
plt.subplot(221), plt.imshow(original)
plt.title('Original Image'), plt.xticks([]),plt.yticks([])

# Threshold
plt.subplot(222), plt.imshow(thresh_original, cmap='gray')
plt.title('threshold binary'), plt.xticks([]),plt.yticks([])

# selected area for future crop
ctr = ctr[:,:,::-1] # flip color for maptolib display
plt.subplot(223), plt.imshow(ctr)
plt.title('Selected area'), plt.xticks([]),plt.yticks([])

plt.show()

结果:

enter image description here


Tags: in图像importimgforthresholdpltcv2