本文整理汇总了Python中pywt.dwtn函数的典型用法代码示例。如果您正苦于以下问题:Python dwtn函数的具体用法?Python dwtn怎么用?Python dwtn使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了dwtn函数的20个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于我们的系统推荐出更棒的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)
开发者ID:liluximax,项目名称:python,代码行数:32,代码来源:test_multidim.py
示例2: test_dwtn_input
def test_dwtn_input():
# Array-like must be accepted
pywt.dwtn([1, 2, 3, 4], 'haar')
# Others must not
data = dict()
assert_raises(TypeError, pywt.dwtn, data, 'haar')
# Must be at least 1D
assert_raises(ValueError, pywt.dwtn, 2, 'haar')
开发者ID:liluximax,项目名称:python,代码行数:8,代码来源:test_multidim.py
示例3: 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)
开发者ID:liluximax,项目名称:python,代码行数:11,代码来源:test_multidim.py
示例4: test_byte_offset
def test_byte_offset():
wavelet = pywt.Wavelet("haar")
for dtype in ("float32", "float64"):
data = np.array([[0, 4, 1, 5, 1, 4], [0, 5, 6, 3, 2, 1], [2, 5, 19, 4, 19, 1]], dtype=dtype)
for mode in pywt.Modes.modes:
expected = pywt.dwtn(data, wavelet)
padded = np.ones((3, 6), dtype=np.dtype([("data", data.dtype), ("pad", "byte")]))
padded[:] = data
padded_dwtn = pywt.dwtn(padded["data"], wavelet)
for key in expected.keys():
assert_allclose(padded_dwtn[key], expected[key])
开发者ID:apetcho,项目名称:pywt,代码行数:12,代码来源:test_multidim.py
示例5: test_stride
def test_stride():
wavelet = pywt.Wavelet("haar")
for dtype in ("float32", "float64"):
data = np.array([[0, 4, 1, 5, 1, 4], [0, 5, 6, 3, 2, 1], [2, 5, 19, 4, 19, 1]], dtype=dtype)
for mode in pywt.Modes.modes:
expected = pywt.dwtn(data, wavelet)
strided = np.ones((3, 12), dtype=data.dtype)
strided[::-1, ::2] = data
strided_dwtn = pywt.dwtn(strided[::-1, ::2], wavelet)
for key in expected.keys():
assert_allclose(strided_dwtn[key], expected[key])
开发者ID:apetcho,项目名称:pywt,代码行数:13,代码来源:test_multidim.py
示例6: 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)
开发者ID:iceseismic,项目名称:odl,代码行数:29,代码来源:wavelet_test.py
示例7: test_dwtn_axes
def test_dwtn_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,))
expected_a = list(map(lambda x: pywt.dwt(x, 'haar')[0], data))
assert_equal(coefs['a'], expected_a)
expected_d = list(map(lambda x: pywt.dwt(x, 'haar')[1], data))
assert_equal(coefs['d'], expected_d)
coefs = pywt.dwtn(data, 'haar', axes=(1, 1))
expected_aa = list(map(lambda x: pywt.dwt(x, 'haar')[0], expected_a))
assert_equal(coefs['aa'], expected_aa)
expected_ad = list(map(lambda x: pywt.dwt(x, 'haar')[1], expected_a))
assert_equal(coefs['ad'], expected_ad)
开发者ID:liluximax,项目名称:python,代码行数:16,代码来源:test_multidim.py
示例8: 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)
开发者ID:liluximax,项目名称:python,代码行数:7,代码来源:test_multidim.py
示例9: test_byte_offset
def test_byte_offset():
wavelet = pywt.Wavelet('haar')
for dtype in ('float32', 'float64'):
data = np.array([[0, 4, 1, 5, 1, 4],
[0, 5, 6, 3, 2, 1],
[2, 5, 19, 4, 19, 1]],
dtype=dtype)
for mode in pywt.Modes.modes:
expected = pywt.dwtn(data, wavelet)
padded = np.ones((3, 6), dtype=np.dtype([('data', data.dtype),
('pad', 'byte')]))
padded[:] = data
padded_dwtn = pywt.dwtn(padded['data'], wavelet)
for key in expected.keys():
assert_allclose(padded_dwtn[key], expected[key])
开发者ID:liluximax,项目名称:python,代码行数:16,代码来源:test_multidim.py
示例10: estimate_sigma
def estimate_sigma(im, average_sigmas=False, multichannel=False):
"""
Robust wavelet-based estimator of the (Gaussian) noise standard deviation.
Parameters
----------
im : ndarray
Image for which to estimate the noise standard deviation.
average_sigmas : bool, optional
If true, average the channel estimates of `sigma`. Otherwise return
a list of sigmas corresponding to each channel.
multichannel : bool
Estimate sigma separately for each channel.
Returns
-------
sigma : float or list
Estimated noise standard deviation(s). If `multichannel` is True and
`average_sigmas` is False, a separate noise estimate for each channel
is returned. Otherwise, the average of the individual channel
estimates is returned.
Notes
-----
This function assumes the noise follows a Gaussian distribution. The
estimation algorithm is based on the median absolute deviation of the
wavelet detail coefficients as described in section 4.2 of [1]_.
References
----------
.. [1] D. L. Donoho and I. M. Johnstone. "Ideal spatial adaptation
by wavelet shrinkage." Biometrika 81.3 (1994): 425-455.
DOI:10.1093/biomet/81.3.425
Examples
--------
>>> import skimage.data
>>> from skimage import img_as_float
>>> img = img_as_float(skimage.data.camera())
>>> sigma = 0.1
>>> img = img + sigma * np.random.standard_normal(img.shape)
>>> sigma_hat = estimate_sigma(img, multichannel=False)
"""
if multichannel:
nchannels = im.shape[-1]
sigmas = [estimate_sigma(
im[..., c], multichannel=False) for c in range(nchannels)]
if average_sigmas:
sigmas = np.mean(sigmas)
return sigmas
elif im.shape[-1] <= 4:
msg = ("image is size {0} on the last axis, but multichannel is "
"False. If this is a color image, please set multichannel "
"to True for proper noise estimation.")
warn(msg.format(im.shape[-1]))
coeffs = pywt.dwtn(im, wavelet='db2')
detail_coeffs = coeffs['d' * im.ndim]
return _sigma_est_dwt(detail_coeffs, distribution='Gaussian')
开发者ID:ameya005,项目名称:scikit-image,代码行数:58,代码来源:_denoise.py
示例11: 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)
开发者ID:j-vibert,项目名称:pywt,代码行数:19,代码来源:test_multidim.py
示例12: 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)
开发者ID:kwohlfahrt,项目名称:pywt,代码行数:12,代码来源:test_multidim.py
示例13: 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)
开发者ID:liluximax,项目名称:python,代码行数:12,代码来源:test_multidim.py
示例14: 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)
开发者ID:apetcho,项目名称:pywt,代码行数:12,代码来源:test_multidim.py
示例15: 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)
开发者ID:PyWavelets,项目名称:pywt,代码行数:14,代码来源:test_multidim.py
示例16: 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)
开发者ID:liluximax,项目名称:python,代码行数:15,代码来源:test_multidim.py
示例17: 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)
开发者ID:apetcho,项目名称:pywt,代码行数:15,代码来源:test_multidim.py
示例18: 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:
if wavelet in ['cmor', 'shan', 'fbsp']:
# skip these CWT families to avoid warnings
continue
if isinstance(pywt.DiscreteContinuousWavelet(wavelet), pywt.Wavelet):
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)
开发者ID:HenryZhou1002,项目名称:pywt,代码行数:16,代码来源:test_multidim.py
示例19: 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")
coefs = pywt.dwtn(data, wavelet)
# No point removing zero, or all
for num_missing in range(1, len(coefs)):
for missing in combinations(coefs.keys(), num_missing):
missing_coefs = coefs.copy()
for key in missing:
del missing_coefs[key]
LL = missing_coefs.get("aa", None)
HL = missing_coefs.get("da", None)
LH = missing_coefs.get("ad", None)
HH = missing_coefs.get("dd", None)
assert_allclose(pywt.idwt2((LL, (HL, LH, HH)), wavelet), pywt.idwtn(missing_coefs, "haar"), atol=1e-15)
开发者ID:apetcho,项目名称:pywt,代码行数:20,代码来源:test_multidim.py
示例20: test_idwtn_axes_subsets
def test_idwtn_axes_subsets():
data = np.array(np.random.standard_normal((4, 4, 4, 4)))
# test all combinations of 3 out of 4 axes transformed
for axes in combinations((0, 1, 2, 3), 3):
coefs = pywt.dwtn(data, 'haar', axes=axes)
assert_allclose(pywt.idwtn(coefs, 'haar', axes=axes), data, atol=1e-14)
开发者ID:liluximax,项目名称:python,代码行数:6,代码来源:test_multidim.py
注:本文中的pywt.dwtn函数示例由纯净天空整理自Github/MSDocs等源码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。 |
请发表评论