本文整理汇总了Python中pywt.idwtn函数的典型用法代码示例。如果您正苦于以下问题:Python idwtn函数的具体用法?Python idwtn怎么用?Python idwtn使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了idwtn函数的15个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_per_axis_wavelets_and_modes
def test_per_axis_wavelets_and_modes():
# tests seperate wavelet and edge mode for each axis.
rstate = np.random.RandomState(1234)
data = rstate.randn(16, 16, 16)
# wavelet can be a string or wavelet object
wavelets = (pywt.Wavelet('haar'), 'sym2', 'db4')
# mode can be a string or a Modes enum
modes = ('symmetric', 'periodization',
pywt._extensions._pywt.Modes.reflect)
coefs = pywt.dwtn(data, wavelets, modes)
assert_allclose(pywt.idwtn(coefs, wavelets, modes), data, atol=1e-14)
coefs = pywt.dwtn(data, wavelets[:1], modes)
assert_allclose(pywt.idwtn(coefs, wavelets[:1], modes), data, atol=1e-14)
coefs = pywt.dwtn(data, wavelets, modes[:1])
assert_allclose(pywt.idwtn(coefs, wavelets, modes[:1]), data, atol=1e-14)
# length of wavelets or modes doesn't match the length of axes
assert_raises(ValueError, pywt.dwtn, data, wavelets[:2])
assert_raises(ValueError, pywt.dwtn, data, wavelets, mode=modes[:2])
assert_raises(ValueError, pywt.idwtn, coefs, wavelets[:2])
assert_raises(ValueError, pywt.idwtn, coefs, wavelets, mode=modes[:2])
# dwt2/idwt2 also support per-axis wavelets/modes
data2 = data[..., 0]
coefs2 = pywt.dwt2(data2, wavelets[:2], modes[:2])
assert_allclose(pywt.idwt2(coefs2, wavelets[:2], modes[:2]), data2,
atol=1e-14)
示例2: test_negative_axes
def test_negative_axes():
data = np.array([[0, 1, 2, 3],
[1, 1, 1, 1],
[1, 4, 2, 8]])
coefs1 = pywt.dwtn(data, 'haar', axes=(1, 1))
coefs2 = pywt.dwtn(data, 'haar', axes=(-1, -1))
assert_equal(coefs1, coefs2)
rec1 = pywt.idwtn(coefs1, 'haar', axes=(1, 1))
rec2 = pywt.idwtn(coefs1, 'haar', axes=(-1, -1))
assert_equal(rec1, rec2)
示例3: test_idwtn_mixed_complex_dtype
def test_idwtn_mixed_complex_dtype():
rstate = np.random.RandomState(0)
x = rstate.randn(8, 8, 8)
x = x + 1j*x
coeffs = pywt.dwtn(x, 'db2')
x_roundtrip = pywt.idwtn(coeffs, 'db2')
assert_allclose(x_roundtrip, x, rtol=1e-10)
# mismatched dtypes OK
coeffs['a' * x.ndim] = coeffs['a' * x.ndim].astype(np.complex64)
x_roundtrip2 = pywt.idwtn(coeffs, 'db2')
assert_allclose(x_roundtrip2, x, rtol=1e-7, atol=1e-7)
assert_(x_roundtrip2.dtype == np.complex128)
示例4: test_idwtn_none_coeffs
def test_idwtn_none_coeffs():
data = np.array([[0, 1, 2, 3],
[1, 1, 1, 1],
[1, 4, 2, 8]])
data = data + 1j*data # test with complex data
coefs = pywt.dwtn(data, 'haar', axes=(1, 1))
# verify setting coefficients to None is the same as zeroing them
coefs['dd'] = np.zeros_like(coefs['dd'])
result_zeros = pywt.idwtn(coefs, 'haar', axes=(1, 1))
coefs['dd'] = None
result_none = pywt.idwtn(coefs, 'haar', axes=(1, 1))
assert_equal(result_zeros, result_none)
示例5: test_wavelet_decomposition3d_and_reconstruction3d
def test_wavelet_decomposition3d_and_reconstruction3d():
# Test 3D wavelet decomposition and reconstruction and verify that
# they perform as expected
x = np.random.rand(16, 16, 16)
mode = 'sym'
wbasis = pywt.Wavelet('db5')
nscales = 1
wavelet_coeffs = wavelet_decomposition3d(x, wbasis, mode, nscales)
aaa = wavelet_coeffs[0]
reference = pywt.dwtn(x, wbasis, mode)
aaa_reference = reference['aaa']
assert all_almost_equal(aaa, aaa_reference)
reconstruction = wavelet_reconstruction3d(wavelet_coeffs, wbasis, mode,
nscales)
reconstruction_reference = pywt.idwtn(reference, wbasis, mode)
assert all_almost_equal(reconstruction, reconstruction_reference)
assert all_almost_equal(reconstruction, x)
assert all_almost_equal(reconstruction_reference, x)
wbasis = pywt.Wavelet('db1')
nscales = 3
wavelet_coeffs = wavelet_decomposition3d(x, wbasis, mode, nscales)
shape_true = (nscales + 1, )
assert all_equal(np.shape(wavelet_coeffs), shape_true)
reconstruction = wavelet_reconstruction3d(wavelet_coeffs, wbasis, mode,
nscales)
assert all_almost_equal(reconstruction, x)
示例6: test_idwtn_axes
def test_idwtn_axes():
data = np.array([[0, 1, 2, 3],
[1, 1, 1, 1],
[1, 4, 2, 8]])
data = data + 1j*data # test with complex data
coefs = pywt.dwtn(data, 'haar', axes=(1, 1))
assert_allclose(pywt.idwtn(coefs, 'haar', axes=(1, 1)), data, atol=1e-14)
示例7: test_ignore_invalid_keys
def test_ignore_invalid_keys():
data = np.array([[0, 4, 1, 5, 1, 4], [0, 5, 6, 3, 2, 1], [2, 5, 19, 4, 19, 1]])
wavelet = pywt.Wavelet("haar")
LL, (HL, LH, HH) = pywt.dwt2(data, wavelet)
d = {"aa": LL, "da": HL, "ad": LH, "dd": HH, "foo": LH, "a": HH}
assert_allclose(pywt.idwt2((LL, (HL, LH, HH)), wavelet), pywt.idwtn(d, wavelet), atol=1e-15)
示例8: test_idwtn_take
def test_idwtn_take():
data = np.array([
[[1, 4, 1, 5, 1, 4],
[0, 5, 6, 3, 2, 1],
[2, 5, 19, 4, 19, 1]],
[[1, 5, 1, 2, 3, 4],
[7, 12, 6, 52, 7, 8],
[5, 2, 6, 78, 12, 2]]])
wavelet = pywt.Wavelet('haar')
d = pywt.dwtn(data, wavelet)
assert_(data.shape != pywt.idwtn(d, wavelet).shape)
assert_allclose(data, pywt.idwtn(d, wavelet, take=data.shape), atol=1e-15)
# Check shape for take not equal to data.shape
data = np.random.randn(51, 17, 68)
d = pywt.dwtn(data, wavelet)
assert_equal((2, 2, 2), pywt.idwtn(d, wavelet, take=2).shape)
assert_equal((52, 18, 68), pywt.idwtn(d, wavelet, take=0).shape)
示例9: test_dwdtn_idwtn_allwavelets
def test_dwdtn_idwtn_allwavelets():
rstate = np.random.RandomState(1234)
r = rstate.randn(16, 16)
# test 2D case only for all wavelet types
wavelist = pywt.wavelist()
if 'dmey' in wavelist:
wavelist.remove('dmey')
for wavelet in wavelist:
for mode in pywt.Modes.modes:
coeffs = pywt.dwtn(r, wavelet, mode=mode)
assert_allclose(pywt.idwtn(coeffs, wavelet, mode=mode),
r, rtol=1e-7, atol=1e-7)
示例10: test_3D_reconstruct
def test_3D_reconstruct():
data = np.array(
[
[[0, 4, 1, 5, 1, 4], [0, 5, 26, 3, 2, 1], [5, 8, 2, 33, 4, 9], [2, 5, 19, 4, 19, 1]],
[[1, 5, 1, 2, 3, 4], [7, 12, 6, 52, 7, 8], [2, 12, 3, 52, 6, 8], [5, 2, 6, 78, 12, 2]],
]
)
wavelet = pywt.Wavelet("haar")
for mode in pywt.Modes.modes:
d = pywt.dwtn(data, wavelet, mode=mode)
assert_allclose(data, pywt.idwtn(d, wavelet, mode=mode), rtol=1e-13, atol=1e-13)
示例11: test_dwtn_idwtn_dtypes
def test_dwtn_idwtn_dtypes():
wavelet = pywt.Wavelet('haar')
for dt_in, dt_out in zip(dtypes_in, dtypes_out):
x = np.ones((4, 4), dtype=dt_in)
errmsg = "wrong dtype returned for {0} input".format(dt_in)
coeffs = pywt.dwtn(x, wavelet)
for k, v in coeffs.items():
assert_(v.dtype == dt_out, "dwtn: " + errmsg)
x_roundtrip = pywt.idwtn(coeffs, wavelet)
assert_(x_roundtrip.dtype == dt_out, "idwtn: " + errmsg)
示例12: test_ignore_invalid_keys
def test_ignore_invalid_keys():
data = np.array([
[0, 4, 1, 5, 1, 4],
[0, 5, 6, 3, 2, 1],
[2, 5, 19, 4, 19, 1]])
wavelet = pywt.Wavelet('haar')
LL, (HL, LH, HH) = pywt.dwt2(data, wavelet)
d = {'aa': LL, 'da': HL, 'ad': LH, 'dd': HH,
'foo': LH, 'a': HH}
assert_allclose(pywt.idwt2((LL, (HL, LH, HH)), wavelet),
pywt.idwtn(d, wavelet), atol=1e-15)
示例13: test_idwtn_missing
def test_idwtn_missing():
# Test to confirm missing data behave as zeroes
data = np.array([
[0, 4, 1, 5, 1, 4],
[0, 5, 6, 3, 2, 1],
[2, 5, 19, 4, 19, 1]])
wavelet = pywt.Wavelet('haar')
LL, (HL, _, HH) = pywt.dwt2(data, wavelet)
d = {'aa': LL, 'da': HL, 'dd': HH}
assert_allclose(pywt.idwt2((LL, (HL, None, HH)), wavelet),
pywt.idwtn(d, 'haar'), atol=1e-15)
示例14: test_3D_reconstruct_complex
def test_3D_reconstruct_complex():
# All dimensions even length so `take` does not need to be specified
data = np.array(
[
[[0, 4, 1, 5, 1, 4], [0, 5, 26, 3, 2, 1], [5, 8, 2, 33, 4, 9], [2, 5, 19, 4, 19, 1]],
[[1, 5, 1, 2, 3, 4], [7, 12, 6, 52, 7, 8], [2, 12, 3, 52, 6, 8], [5, 2, 6, 78, 12, 2]],
]
)
data = data + 1j
wavelet = pywt.Wavelet("haar")
d = pywt.dwtn(data, wavelet)
# idwtn creates even-length shapes (2x dwtn size)
original_shape = [slice(None, s) for s in data.shape]
assert_allclose(data, pywt.idwtn(d, wavelet)[original_shape], rtol=1e-13, atol=1e-13)
示例15: test_idwtn_idwt2_complex
def test_idwtn_idwt2_complex():
data = np.array([
[0, 4, 1, 5, 1, 4],
[0, 5, 6, 3, 2, 1],
[2, 5, 19, 4, 19, 1]])
data = data + 1j
wavelet = pywt.Wavelet('haar')
LL, (HL, LH, HH) = pywt.dwt2(data, wavelet)
d = {'aa': LL, 'da': HL, 'ad': LH, 'dd': HH}
for mode in pywt.Modes.modes:
assert_allclose(pywt.idwt2((LL, (HL, LH, HH)), wavelet, mode=mode),
pywt.idwtn(d, wavelet, mode=mode),
rtol=1e-14, atol=1e-14)