提交 5610f9d4 authored 作者: amrithasuresh's avatar amrithasuresh

Updated numpy as np

上级 786a9fa7
...@@ -4,7 +4,7 @@ Provides Ops for FFT and DCT. ...@@ -4,7 +4,7 @@ Provides Ops for FFT and DCT.
""" """
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import numpy import numpy as np
import numpy.fft import numpy.fft
from six.moves import xrange from six.moves import xrange
...@@ -126,13 +126,13 @@ def dct_matrix(rows, cols, unitary=True): ...@@ -126,13 +126,13 @@ def dct_matrix(rows, cols, unitary=True):
This algorithm is adapted from Dan Ellis' Rastmat spec2cep.m, lines 15-20. This algorithm is adapted from Dan Ellis' Rastmat spec2cep.m, lines 15-20.
""" """
rval = numpy.zeros((rows, cols)) rval = np.zeros((rows, cols))
col_range = numpy.arange(cols) col_range = np.arange(cols)
scale = numpy.sqrt(2.0 / cols) scale = np.sqrt(2.0 / cols)
for i in xrange(rows): for i in xrange(rows):
rval[i] = numpy.cos( rval[i] = np.cos(
i * (col_range * 2 + 1) / (2.0 * cols) * numpy.pi) * scale i * (col_range * 2 + 1) / (2.0 * cols) * np.pi) * scale
if unitary: if unitary:
rval[0] *= numpy.sqrt(0.5) rval[0] *= np.sqrt(0.5)
return rval return rval
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论