├── .gitignore ├── LICENSE ├── README.md ├── erf.py ├── images ├── ERF.jpg └── Example.jpg ├── logs ├── val_base.out ├── val_small.out └── val_tiny.out ├── outputs_erf └── pre_computed │ ├── RepLK-T_31.npy │ ├── SLaK-T_51.npy │ └── WTConvNext-T_5.npy ├── semantic_segmentation ├── README.md ├── configs │ └── wtconvnext │ │ ├── upernet_wtconvnext.py │ │ ├── wtconvnext-base_upernet_8xb2-amp-160k_ade20k-512x512.py │ │ ├── wtconvnext-small_upernet_8xb2-amp-160k_ade20k-512x512.py │ │ └── wtconvnext-tiny_upernet_8xb2-amp-160k_ade20k-512x512.py ├── mmseg │ └── models │ │ └── backbones │ │ └── wtconvnext.py └── translate_model_to_mmseg.py ├── train.py ├── validate.py ├── wtconv ├── __init__.py ├── util │ ├── __init__.py │ └── wavelet.py ├── wtconv1d.py └── wtconv2d.py └── wtconvnext ├── __init__.py └── wtconvnext.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/README.md -------------------------------------------------------------------------------- /erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/erf.py -------------------------------------------------------------------------------- /images/ERF.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/images/ERF.jpg -------------------------------------------------------------------------------- /images/Example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/images/Example.jpg -------------------------------------------------------------------------------- /logs/val_base.out: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/logs/val_base.out -------------------------------------------------------------------------------- /logs/val_small.out: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/logs/val_small.out -------------------------------------------------------------------------------- /logs/val_tiny.out: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/logs/val_tiny.out -------------------------------------------------------------------------------- /outputs_erf/pre_computed/RepLK-T_31.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/outputs_erf/pre_computed/RepLK-T_31.npy -------------------------------------------------------------------------------- /outputs_erf/pre_computed/SLaK-T_51.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/outputs_erf/pre_computed/SLaK-T_51.npy -------------------------------------------------------------------------------- /outputs_erf/pre_computed/WTConvNext-T_5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/outputs_erf/pre_computed/WTConvNext-T_5.npy -------------------------------------------------------------------------------- /semantic_segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/README.md -------------------------------------------------------------------------------- /semantic_segmentation/configs/wtconvnext/upernet_wtconvnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/configs/wtconvnext/upernet_wtconvnext.py -------------------------------------------------------------------------------- /semantic_segmentation/configs/wtconvnext/wtconvnext-base_upernet_8xb2-amp-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/configs/wtconvnext/wtconvnext-base_upernet_8xb2-amp-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /semantic_segmentation/configs/wtconvnext/wtconvnext-small_upernet_8xb2-amp-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/configs/wtconvnext/wtconvnext-small_upernet_8xb2-amp-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /semantic_segmentation/configs/wtconvnext/wtconvnext-tiny_upernet_8xb2-amp-160k_ade20k-512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/configs/wtconvnext/wtconvnext-tiny_upernet_8xb2-amp-160k_ade20k-512x512.py -------------------------------------------------------------------------------- /semantic_segmentation/mmseg/models/backbones/wtconvnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/mmseg/models/backbones/wtconvnext.py -------------------------------------------------------------------------------- /semantic_segmentation/translate_model_to_mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/semantic_segmentation/translate_model_to_mmseg.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/train.py -------------------------------------------------------------------------------- /validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/validate.py -------------------------------------------------------------------------------- /wtconv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/wtconv/__init__.py -------------------------------------------------------------------------------- /wtconv/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /wtconv/util/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/wtconv/util/wavelet.py -------------------------------------------------------------------------------- /wtconv/wtconv1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/wtconv/wtconv1d.py -------------------------------------------------------------------------------- /wtconv/wtconv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/wtconv/wtconv2d.py -------------------------------------------------------------------------------- /wtconvnext/__init__.py: -------------------------------------------------------------------------------- 1 | from .wtconvnext import * 2 | -------------------------------------------------------------------------------- /wtconvnext/wtconvnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BGU-CS-VIL/WTConv/HEAD/wtconvnext/wtconvnext.py --------------------------------------------------------------------------------