Tensorflow wavelet Layers
Project description
tensorflow-wavelets is an implementation of
- Discrete Wavelets Transform Layer
- Duel Tree Complex Wavelets Transform Layer
- Multi Wavelets Transform Neural Networks Layer
Installation
pip install tensorflow-wavelets
Usage
import tensorflow_wavelets.Layers.DWT as DWT
import tensorflow_wavelets.Layers.DTCWT as DTCWT
import tensorflow_wavelets.Layers.DMWT as DMWT
Example
model = keras.Sequential()
model.add(keras.Input(shape=input_shape))
model.add(DWT.DWT(name="haar",concat=0))
# name can be found in pywt.wavelist(family)
# concat=0 means to split to 4 smaller layers
# concat=1 will output 1 big layer - concat from 4 smaller layers
model.add(keras.layers.Dropout(0.5))
model.add(keras.layers.Flatten())
model.add(keras.layers.Dense(nb_classes, activation="softmax"))
model.summary()
Free Software, Hell Yeah!
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tensorflow-wavelets-1.0.17.tar.gz
(16.8 kB
view hashes)
Built Distribution
Close
Hashes for tensorflow-wavelets-1.0.17.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 67f45daa80cf3fd97134fb7d884929ef4cc74389f6ea93213fbe0908110ae873 |
|
MD5 | f3df40c7f7619c6ddd3aa519e0f36384 |
|
BLAKE2b-256 | f58a0536a453655d49c47b51098034e60334b1a473c15b4a71d112d9771d30ea |
Close
Hashes for tensorflow_wavelets-1.0.17-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3b51cb11ce7b539c824a02fdeaab7a00591bc0c23b0dd219cf006b637865224f |
|
MD5 | f02222906f5cb5933189321538a0e298 |
|
BLAKE2b-256 | 75d4634620883573a34ffd210307e2d3535f459ae0ddc9a7c1af31b5bf2baf8b |