在Python中删除图像的边框

2024-05-10 03:34:10 发布

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

有些视频的帧有黑色的条纹,如边框。我得把它们从镜框里拿出来。我想出了一个粗略的解决方案:

import sys, cv2, numpy
import Image, scipy

filename = "snap.jpeg"

img = cv2.imread(filename)

def checkEqual(lst):
    return len(set(lst)) <= 1 ## <-- This is the maximum length of the set

def removeColumns(image):
    for col in range(image.shape[1]):
        for ch in range(3):
            try:
                checkEqual(image[:, col, ch].tolist())
            except IndexError:
                continue
            else:
                if checkEqual(image[:, col, ch].tolist()):
                    try:
                        image = numpy.delete(image, col, 1)
                    except IndexError:
                        continue
                    else:
                        pass
    return image

img2 = removeColumns(img)

print img.shape, img2.shape ## (480, 856, 3) (480, 705, 3)

在这里我找到了具有相同元素的列和所有具有黑色边框的视频。但即使我将函数checkEqual()中的最大长度从1增加到20或40,也不会删除整个黑色条带。

这是原始图像: enter image description here

这是运行程序后的图像: enter image description here

有谁能提出更好地解决这个问题的建议吗? 谢谢!


Tags: imageimportnumpyimg视频defcolch
3条回答

为什么不计算框架并使用PIL

from PIL import Image

img = Image.open('myImage.jpeg')
box = (50, 50, 100, 100)
area = img.crop(box)

这个问题已经在this answer中解决了。

In [1]: from PIL import Image, ImageChops

In [3]: im = Image.open('iI3ZE.jpg')

In [4]: def trim(im):
   ...:         bg = Image.new(im.mode, im.size, im.getpixel((0,0)))
   ...:         diff = ImageChops.difference(im, bg)
   ...:         diff = ImageChops.add(diff, diff, 2.0, -100)
   ...:         bbox = diff.getbbox()
   ...:         if bbox:
   ...:                 return im.crop(bbox)
   ...:

In [5]: trim(im).show()

我用枕头代替棉花:

pip install pillow

结果:

enter image description here

在尝试使用opencv和numpy时,类似这样的方法如何:

im = cv2.imread(filename)
h,w,d = im.shape
#left limit
for i in range(w):
    if np.sum(im[:,i,:]) > 0:
        break
#right limit
for j in xrange(w-1,0,-1):
    if np.sum(im[:,j,:]) > 0:
        break

cropped = im[:,i:j+1,:].copy() # deep copy to get byte-aligned array needed for opencv 

相关问题 更多 >