├── README.md ├── eval.py ├── eval.sh ├── figures ├── context.png ├── results.png └── structure.png ├── models ├── MambaIC.py ├── VSS_module.py ├── __init__.py └── csm_triton.py ├── requirements.txt ├── train.py └── train.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/README.md -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/eval.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/eval.sh -------------------------------------------------------------------------------- /figures/context.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/figures/context.png -------------------------------------------------------------------------------- /figures/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/figures/results.png -------------------------------------------------------------------------------- /figures/structure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/figures/structure.png -------------------------------------------------------------------------------- /models/MambaIC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/models/MambaIC.py -------------------------------------------------------------------------------- /models/VSS_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/models/VSS_module.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/csm_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/models/csm_triton.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | compressai 2 | torch=2.1.0 3 | einops 4 | mamba-ssm 5 | timm 6 | tqdm -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AuroraZengfh/MambaIC/HEAD/train.sh --------------------------------------------------------------------------------