前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python 3D-DCT

Python 3D-DCT

作者头像
Aidol
发布2021-05-11 11:44:51
7370
发布2021-05-11 11:44:51
举报
文章被收录于专栏:苦旅苦旅
代码语言:javascript
复制
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)
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档