from scipy.fftpack import dct,idct
import numpy as np
def dct3(gop):
'''
gop shape is bath*h*w
'''
assert type(gop) == np.ndarray,"输入必须为numpy对象"
gop_dct = []
# 先对图像做二维的dct变换
for frame in gop:
frame_dct2 = dct(dct(frame.astype(np.double), norm='ortho').T,norm='ortho').T
gop_dct.append(frame_dct2)
gop_dct = np.array(gop_dct)
result = np.zeros_like(gop_dct)
_,h,w = gop.shape
# 对帧间做dct变换
for i in range(h):
for j in range(w):
result[:,i,j] = dct(gop_dct[:,i,j],norm='ortho')
return result
def idct3(gop_dct):
'''
gop_dct shape is bath*h*w
'''
assert type(gop_dct) == np.ndarray,"输入必须为numpy对象"
# 先对帧间做idct
_,h,w = gop_dct.shape
frame_rec = np.zeros_like(gop_dct)
for i in range(h):
for j in range(w):
frame_rec[:,i,j] = idct(gop_dct[:,i,j],norm='ortho')
result = []
for frame in frame_rec:
result.append(idct(idct(frame, norm='ortho').T,norm='ortho').T)
return np.array(result)
Python 3D-DCT
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- Project 3 – 3D RenderingOverviewFor this project, you are...