RGB image?

[ng0323]

Hi, it seems to work for grayscale images only? I get an error for RGB images.

[nahliabdelwahed]

is it really can support RGB image ?

[zzhuolun]

@ng0323
the error about RGB images is due to that convolutions are done through scipy.signal.conv2d, which only takes in inputs (kernel and image) of 2 dimensions. RGB images have 3 dimensions so cannot be processed by scipy.signal.conv2d.
I used cv2.filter2D to due with RGB imges. So e.g, changing the LinearMotionBlur(img, dim, angle, linetype) in pyblur/LinearMotionBlur.py as follows can do motion blur with RGB images:

def LinearMotionBlur(img, dim, angle, linetype):
    imgarray = np.array(img)
    imgarray=cv2.cvtColor(imgarray, cv2.COLOR_RGB2BGR)
    kernel = LineKernel(dim, angle, linetype)
    convolved=cv2.filter2D(imgarray, -1, kernel)
    convolved = cv2.cvtColor(convolved, cv2.COLOR_BGR2RGB)
    # convolved = convolve2d(imgarray, kernel, mode='same', fillvalue=255.0).astype("uint8")
    img = Image.fromarray(convolved)
    return img