├── .gitignore ├── LICENSE ├── README.md ├── cfg ├── high_rate_model.yaml └── low_rate_model.yaml ├── evaluate.py ├── models ├── __init__.py ├── envc.py ├── layers.py ├── layers_entropy.py └── layers_warp.py ├── requirements.txt └── third_party ├── __init__.py └── msda ├── __init__.py ├── functions ├── __init__.py └── ms_deform_attn_func.py ├── make.sh ├── modules ├── __init__.py └── ms_deform_attn.py ├── setup.py ├── src ├── cpu │ ├── ms_deform_attn_cpu.cpp │ └── ms_deform_attn_cpu.h ├── cuda │ ├── ms_deform_attn_cuda.cu │ ├── ms_deform_attn_cuda.h │ └── ms_deform_im2col_cuda.cuh ├── ms_deform_attn.h └── vision.cpp └── test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/README.md -------------------------------------------------------------------------------- /cfg/high_rate_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/cfg/high_rate_model.yaml -------------------------------------------------------------------------------- /cfg/low_rate_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/cfg/low_rate_model.yaml -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/evaluate.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .envc import * 2 | -------------------------------------------------------------------------------- /models/envc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/models/envc.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/layers_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/models/layers_entropy.py -------------------------------------------------------------------------------- /models/layers_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/models/layers_warp.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pyyaml 2 | pytorch-msssim 3 | easydict -------------------------------------------------------------------------------- /third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/msda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/msda/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/functions/__init__.py -------------------------------------------------------------------------------- /third_party/msda/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /third_party/msda/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/make.sh -------------------------------------------------------------------------------- /third_party/msda/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/modules/__init__.py -------------------------------------------------------------------------------- /third_party/msda/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /third_party/msda/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/setup.py -------------------------------------------------------------------------------- /third_party/msda/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /third_party/msda/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /third_party/msda/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /third_party/msda/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /third_party/msda/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /third_party/msda/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/ms_deform_attn.h -------------------------------------------------------------------------------- /third_party/msda/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/src/vision.cpp -------------------------------------------------------------------------------- /third_party/msda/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-IMCL/ENVC/HEAD/third_party/msda/test.py --------------------------------------------------------------------------------