如何在 Python 中的 OpenCV2 中獲取彩色圖像的接縫？

Question

我正在編寫一些代碼以允許我從圖像中刪除接縫。 目前，我的代碼允許我從灰度圖像中查找、突出顯示和刪除接縫。 我需要從彩色圖像中去除接縫，而我的代碼不允許我這樣做。 任何人都可以幫我修改我的代碼以允許我這樣做嗎？

我的代碼：

import numpy as np
import cv2
import math
import time

def getEdgeImage(img,margin=10):
    kernel=np.float64([[-1,0,1]])
    Ix=cv2.filter2D(img,cv2.CV_64F,kernel)
    Iy=cv2.filter2D(img,cv2.CV_64F,kernel)
    I=np.hypot(Ix,Iy)
    m=I.max()
    I[:,:margin]=m
    I[:,-margin:]=m
    return I

def getEnergyMap(img,repulseMask=None,attractMask=None):
    edges=getEdgeImage(img)
    if attractMask is not None:
        edges[attractMask==1]=-10
    if repulseMask is not None:
        edges[repulseMask==1]=235
    kernel=np.ones(3,np.float64)
    for i in range(1,len(edges)):
        minAbove=cv2.erode(edges[i-1],kernel).T[0]
        edges[i]+=minAbove
    return edges

def getSeam(img,repulseMask=None,attractMask=None):
    energyMap=getEnergyMap(img,repulseMask,attractMask)
    y=len(energyMap)-1
    x=np.argmin(energyMap[y])
    seam=[(x,y)]
    while len(seam)<len(energyMap):
        x,y=seam[-1]
        newY=y-1
        newX=x+np.argmin(energyMap[newY,x-1:x+2])-1
        seam.append((newX,newY))
    return seam

img1=cv2.imread("image.jpg") #[::2,::2]
# attractMask=img1*0
# repulseMask=img1*0
seam=getSeam(img1)

吸引和排斥蒙版對當前的代碼並不重要，它們只是被使用，因此我可以手動插入像素坐標以增加或減少該坐標平面中的接縫數量。

運行此代碼時出現的錯誤：

Traceback (most recent call last):
  File "Program.py", line 110, in <module>
    seam=getSeam(img1)
  File "Program.py", line 62, in getSeam
    energyMap=getEnergyMap(img,repulseMask,attractMask)
  File "Program.py", line 58, in getEnergyMap
    edges[i]+=minAbove
ValueError: operands could not be broadcast together with shapes (960,3) (960,) (960,3)

無論如何，我可以讓它與我的代碼一起工作嗎？ 如果這是我需要做的，我會修改這些功能。

Answer 1

然后試試這個，這些是單獨給函數的單獨通道。

r=img1[:,:,0]
seam_r=getSeam(r)

g=img1[:,:,1]
seam_g=getSeam(g)

b=img1[:,:,2]
seam_b=getSeam(b)

在此之后，將結果單獨傳遞給您的 post 函數。