有些视频的框架有黑色条状边框.我必须从框架中删除它们.我想出了一个粗糙的解决方案:
import sys, cv2, numpy
import Image, scipy
filename = "snap.jpeg"
img = cv2.imread(filename)
def checkEqual(lst):
return len(set(lst)) <= 1 ##
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,也不会删除整个黑条.
这是原始图片:
这是运行程序后的图像:
任何人都可以建议更好地解决这个问题吗?
谢谢!