Metadata-Version: 2.4
Name: wavenetr
Version: 0.0.1
Summary: Wavelet-band transformer segmentation networks
Project-URL: Homepage, https://github.com/kkt-ee/wavenetr
Project-URL: Issues, https://github.com/kkt-ee/wavenetr/issues
Author-email: Kishore Kumar Tarafdar <kishorektarafdar@gmail.com>
License-Expression: Apache-2.0
License-File: LICENSE
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering
Requires-Python: >=3.9
Requires-Dist: tensorflow>=2.13
Requires-Dist: waveletvit>=0.0.1
Description-Content-Type: text/markdown

# wavenetr: Wavelet-Band Transformer Segmentation Networks

[![PyPI Version](https://img.shields.io/pypi/v/wavenetr?label=PyPI&color=gold)](https://pypi.org/project/wavenetr/)
[![Python Versions](https://img.shields.io/pypi/pyversions/wavenetr)](https://pypi.org/project/wavenetr/)
[![TensorFlow](https://img.shields.io/badge/tensorflow-required-darkorange)](https://www.tensorflow.org/)
[![License](https://img.shields.io/badge/license-Apache--2.0-deepgreen.svg?style=flat)](LICENSE)

`wavenetr` provides TensorFlow/Keras WaveNETR models for 2D and 3D segmentation using WaveletViT bottleneck blocks.

## Capabilities

- `WaveNETR2D` and `WaveNETR3D` segmentation models.
- `WaveNETR2DSquareCore` and `WaveNETR3DSquareCore` for fixed square/cubic inputs.
- `CVWaveNETR2D` and `CVWaveNETR3D` variational variants.
- Channel-last TensorFlow/Keras inputs.

## Limitations

- Square inputs are required for `WaveNETR2DSquareCore`.
- Cubic inputs are required for `WaveNETR3DSquareCore`.
- Spatial sizes must be compatible with the configured wavelet levels and token stride.

## Dependencies

- `tensorflow`
- `waveletvit`

## Installation

```bash
pip install wavenetr
```

## Minimal Example

```python
import tensorflow as tf
from wavenetr import WaveNETR3D

model = WaveNETR3D(
    input_shape=(32, 32, 32, 1),
    n_classes=4,
)

x = tf.random.normal([1, 32, 32, 32, 1])
y = model(x)

print(y.shape)
```

## Citation

This software is released for broad research, educational, and engineering use. If this package helps your work, please cite the following paper:

```bibtex
@misc{tarafdar2026interpretablefrugallearningsystems,
      title={Interpretable and Frugal Learning Systems Employing Multiresolution Pyramids and Volterra Kernels},
      author={Kishore Kumar Tarafdar},
      year={2026},
      eprint={2606.15011},
      archivePrefix={arXiv},
      primaryClass={eess.SP},
      url={https://arxiv.org/abs/2606.15011},
}
```

## License

Apache License 2.0. See [`LICENSE`](LICENSE).

Copyright 2026 Kishore Kumar Tarafdar.
