Question

我正在尝试使用OpenCV 2.3 Python包装器来计算图像的DCT。据说，图像== numpy arrays == CV矩阵，所以我觉得这应该有用：

import cv2
img1 = cv2.imread('myimage.jpg', cv2.CV_LOAD_IMAGE_GRAYSCALE)
img2 = cv2.dct(img1)

然而，这会引发错误：

cv2.error: /usr/local/lib/OpenCV-2.3.1/modules/core/src/dxt.cpp:2247: error: (-215) type == CV_32FC1 || type == CV_64FC1 in function dct

我意识到错误意味着输入应该是32位或64位单通道浮点矩阵。但是，我认为在指定灰度时我的图像应该是如何加载的，或者至少它应该足够接近以便CV2能够找出转换。

使用cv2转换DCT图像的适当方法是什么？

Answer 1

这是我从openCV论坛获得的一个解决方案，它有效。

img = cv2.imread(fn, 0)      # 1 chan, grayscale!
imf = np.float32(img)/255.0  # float conversion/scale
dst = cv2.dct(imf)           # the dct
img = np.uint8(dst)*255.0    # convert back

Answer 2

好吧，当您将图像加载为灰度时，它实际上是以每像素8位读取而不是32位浮点值。

您将如何做到这一点：

img1_32f = cv.CreateImage( cv.GetSize(img1), cv.IPL_DEPTH_64F, 1)
cv.Scale(img1, img1_32f, 1.0, 0.0)

另外，请查看dft.py示例。这应该让您了解如何使用dft。

Answer 3

使用cv2似乎没有任何简单的方法。我能找到的最接近的解决方案是：

import cv, cv2
import numpy as np

img1 = cv2.imread('myimage.jpg', cv2.CV_LOAD_IMAGE_GRAYSCALE)
h, w = img1.shape[:2]
vis0 = np.zeros((h,w), np.float32)
vis0[:h, :w] = img1
vis1 = cv2.dct(vis0)
img2 = cv.CreateMat(vis1.shape[0], vis1.shape[1], cv.CV_32FC3)
cv.CvtColor(cv.fromarray(vis1), img2, cv.CV_GRAY2BGR)

cv.SaveImage('output.jpg', img2)

Answer 4

Numpy拥有切片运算符，可以在不同顺序的数组之间工作。

import cv2
import cv2.cv as cv
import numpy as np   

img1 = cv2.imread('myimage.jpg')
# or use cv2.CV_LOAD_IMAGE_GRAYSCALE 
img1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
cv2.imshow('input', img1)
w,h = img1.shape
# make a 32bit float for doing the dct within
img2 = np.zeros((w,h), dtype=np.float32)
print img1.shape, img2.shape
img2 = img2+img1[:w, :h]
dct1 = cv2.dct(img2)
key = -1
while(key < 0):
    cv2.imshow("DCT", dct1)
    key = cv2.waitKey(1)
cv2.destroyAllWindows()

Answer 5

我不想写这个答案，但是当我看到一些答案被选错了时，我决定写。

dct操作可在任何范围的输入上使用，因此我真的不明白为什么其他人将其缩放为[0，1]。但是在opencv中，您需要传递numpy.float32数字。

x = np.array([8, 16, 24 , 32, 40, 48, 56, 64])
cv2.dct(np.float32(x))

# output
array([[ 101.82337189],
       [ -51.53858566],
       [   0.        ],
       [  -5.38763857],
       [   0.        ],
       [  -1.60722351],
       [   0.        ],
       [  -0.40561762]], dtype=float32)

但是如果缩放它，几乎所有的小值都会丢失。

以下是公式和示例的链接： https://users.cs.cf.ac.uk/Dave.Marshall/Multimedia/node231.html#DCTbasis

Answer 6

在这里如何使用scipy：

import os.path
import numpy as np
from PIL import Image
from scipy.fftpack import fft, dct

if __name__ == '__main__':
    image_counter = 1

    # Apply DCT to the noisy image patches.
    noise_image_path = 'noise_images/' + str(image_counter) + '.png'
    noise_image = Image.open(noise_image_path)         
    noise_dct_data = dct(noise_image)
    print(noise_dct_data)

用OpenCV计算图像的离散余弦变换

6 个答案: