├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── alexnet.pth ├── arguments └── __init__.py ├── configs └── MambaGRU.py ├── datasets └── stcgs_dataset.py ├── flow ├── core │ ├── __init__.py │ ├── corr.py │ ├── datasets.py │ ├── extractor.py │ ├── raft.py │ ├── update.py │ └── utils │ │ ├── __init__.py │ │ ├── augmentor.py │ │ ├── flow_viz.py │ │ ├── frame_utils.py │ │ └── utils.py └── models │ ├── raft-small.pth │ └── raft-things.pth ├── models └── MambaGRU.py ├── mp_represent.py ├── radar_gs ├── camera.py ├── gaussian_model.py ├── radar_dataset.py └── render.py ├── requirements.txt ├── test.py ├── train.py └── utils ├── general_utils.py ├── graphics_utils.py ├── loss_utils.py ├── lpips_radar.py ├── metrics.py ├── preprocess.py ├── sort_stat.py └── system_utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .vscode/ 2 | __pycache__/ 3 | outputs/ -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/README.md -------------------------------------------------------------------------------- /alexnet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/alexnet.pth -------------------------------------------------------------------------------- /arguments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/arguments/__init__.py -------------------------------------------------------------------------------- /configs/MambaGRU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/configs/MambaGRU.py -------------------------------------------------------------------------------- /datasets/stcgs_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/datasets/stcgs_dataset.py -------------------------------------------------------------------------------- /flow/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow/core/corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/corr.py -------------------------------------------------------------------------------- /flow/core/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/datasets.py -------------------------------------------------------------------------------- /flow/core/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/extractor.py -------------------------------------------------------------------------------- /flow/core/raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/raft.py -------------------------------------------------------------------------------- /flow/core/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/update.py -------------------------------------------------------------------------------- /flow/core/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow/core/utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/utils/augmentor.py -------------------------------------------------------------------------------- /flow/core/utils/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/utils/flow_viz.py -------------------------------------------------------------------------------- /flow/core/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/utils/frame_utils.py -------------------------------------------------------------------------------- /flow/core/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/core/utils/utils.py -------------------------------------------------------------------------------- /flow/models/raft-small.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/models/raft-small.pth -------------------------------------------------------------------------------- /flow/models/raft-things.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/flow/models/raft-things.pth -------------------------------------------------------------------------------- /models/MambaGRU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/models/MambaGRU.py -------------------------------------------------------------------------------- /mp_represent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/mp_represent.py -------------------------------------------------------------------------------- /radar_gs/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/radar_gs/camera.py -------------------------------------------------------------------------------- /radar_gs/gaussian_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/radar_gs/gaussian_model.py -------------------------------------------------------------------------------- /radar_gs/radar_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/radar_gs/radar_dataset.py -------------------------------------------------------------------------------- /radar_gs/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/radar_gs/render.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/train.py -------------------------------------------------------------------------------- /utils/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/general_utils.py -------------------------------------------------------------------------------- /utils/graphics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/graphics_utils.py -------------------------------------------------------------------------------- /utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/loss_utils.py -------------------------------------------------------------------------------- /utils/lpips_radar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/lpips_radar.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/preprocess.py -------------------------------------------------------------------------------- /utils/sort_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/sort_stat.py -------------------------------------------------------------------------------- /utils/system_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ziyeeee/STC-GS/HEAD/utils/system_utils.py --------------------------------------------------------------------------------