Question

我试图将赵科赫的隐写方法从matlab重写为python，而我在开始时就被卡住了。

matlab中的前两个程序：

第1步：

A = imread(casepath); # Reading stegonography case image and aquiring it's RGB values. In my case it's a 400x400 PNG image, so it gives a 400x400x3 array.

第2步：

D = dct2(A(:,:,3)); # Applying 2D DCT to blue values of the image

Python代码模拟：

from scipy import misc
from numpy import empty,arange,exp,real,imag,pi
from numpy.fft import rfft,irfft

arr = misc.imread('casepath')# 400x480x3 array (Step 1)
arr[20, 30, 2] # Getting blue pixel value

def dct(y): #Basic DCT build from numpy
    N = len(y)
    y2 = empty(2*N,float)
    y2[:N] = y[:]
    y2[N:] = y[::-1]

    c = rfft(y2)
    phi = exp(-1j*pi*arange(N)/(2*N))
    return real(phi*c[:N])


def dct2(y): #2D DCT bulid from numpy and using prvious DCT function
    M = y.shape[0]
    N = y.shape[1]
    a = empty([M,N],float)
    b = empty([M,N],float)

    for i in range(M):
        a[i,:] = dct(y[i,:])
    for j in range(N):
        b[:,j] = dct(a[:,j])

    return b

 D = dct2(arr) # step 2 anlogue

但是，当我尝试执行代码时，我收到以下错误：

Traceback (most recent call last):
File "path to .py file", line 31, in <module>
D = dct2(arr)
File "path to .py file", line 25, in dct2
a[i,:] = dct(y[i,:])
File "path to .py file", line 10, in dct
y2[:N] = y[:]
ValueError: could not broadcast input array from shape (400,3) into shape (400)

也许有人可以向我解释我做错了什么？

其他信息：操作系统：Windows 10 Pro 64位 Python：2.7.12 SciPy的：0.18.1 numpy的：1.11.2 枕头：3.4.1

Answer 1

您的代码工作正常，但它只能接受2D数组，就像Matlab中的dct2()一样。由于您的arr是3D数组，因此您需要执行

D = dct2(arr[...,2])

正如我在评论中所提到的，改为使用scipy包中的（快速）内置dct()来代替或重新发明轮子。

我评论中链接的代码有效地为您提供了这个：

import numpy as np
from scipy.fftpack import dct, idct

def dct2(block):
    return dct(dct(block.T, norm='ortho').T, norm='ortho')

def idct2(block):
    return idct(idct(block.T, norm='ortho').T, norm='ortho')

但同样，我必须强调你必须为每个颜色平面分别调用此函数。 Scipy的dct()将很乐意接受任何N维数组，并将在最后一个轴上应用变换。既然这是您的颜色平面而不是像素的行和列，您将得到错误的结果。是的，有一种方法可以使用axis输入参数解决这个问题，但我不会过分复杂化这个答案。

关于此处涉及的各种DCT实现，如果从上面的代码段中省略norm='ortho'参数，则您的版本和scipy的实现会给出相同的结果。但是如果包含该参数，scipy的变换将与Matlab的一致。

在Python中实现2D离散余弦变换的Issiue

1 个答案: