├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── nuscenes_pretrain.py │ │ └── nuscenes_seg.py │ ├── default_runtime.py │ ├── models │ │ ├── minkunet.py │ │ └── superflow.py │ └── schedules │ │ ├── downstream.py │ │ └── pretrain.py ├── downstream │ ├── minkunet34_1pct.py │ └── minkunet34_lp.py └── pretrain │ └── superflow_vit-b_minkunet34.py ├── dist_train.sh ├── docs ├── DATA_PREPAER.md ├── GET_STARTED.md ├── INSTALL.md └── figs │ └── superflow.png ├── setup.cfg ├── superflow ├── __init__.py ├── datasets │ ├── __init__.py │ ├── nuscenes_dataset.py │ └── transforms │ │ ├── __init__.py │ │ ├── formatting.py │ │ ├── loading.py │ │ └── transforms.py └── models │ ├── __init__.py │ ├── backbones │ ├── __init__.py │ ├── dinov2 │ │ ├── __init__.py │ │ ├── dinov2_vision_transformer.py │ │ └── layers │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── block.py │ │ │ ├── dino_head.py │ │ │ ├── drop_path.py │ │ │ ├── layer_scale.py │ │ │ ├── mlp.py │ │ │ ├── patch_embed.py │ │ │ └── swiglu_ffn.py │ ├── minkunet.py │ └── vit.py │ ├── data_preprocessors │ ├── __init__.py │ └── data_preprocessor.py │ ├── seg_heads │ ├── __init__.py │ ├── linear_head.py │ └── upsample_head.py │ ├── segmentors │ ├── __init__.py │ └── voxel_segmentor.py │ └── selfsup │ ├── __init__.py │ └── superflow.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/nuscenes_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/datasets/nuscenes_pretrain.py -------------------------------------------------------------------------------- /configs/_base_/datasets/nuscenes_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/datasets/nuscenes_seg.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/minkunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/models/minkunet.py -------------------------------------------------------------------------------- /configs/_base_/models/superflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/models/superflow.py -------------------------------------------------------------------------------- /configs/_base_/schedules/downstream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/schedules/downstream.py -------------------------------------------------------------------------------- /configs/_base_/schedules/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/_base_/schedules/pretrain.py -------------------------------------------------------------------------------- /configs/downstream/minkunet34_1pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/downstream/minkunet34_1pct.py -------------------------------------------------------------------------------- /configs/downstream/minkunet34_lp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/downstream/minkunet34_lp.py -------------------------------------------------------------------------------- /configs/pretrain/superflow_vit-b_minkunet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/configs/pretrain/superflow_vit-b_minkunet34.py -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/dist_train.sh -------------------------------------------------------------------------------- /docs/DATA_PREPAER.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/docs/DATA_PREPAER.md -------------------------------------------------------------------------------- /docs/GET_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/docs/GET_STARTED.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/figs/superflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/docs/figs/superflow.png -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/setup.cfg -------------------------------------------------------------------------------- /superflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /superflow/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/__init__.py -------------------------------------------------------------------------------- /superflow/datasets/nuscenes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/nuscenes_dataset.py -------------------------------------------------------------------------------- /superflow/datasets/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/transforms/__init__.py -------------------------------------------------------------------------------- /superflow/datasets/transforms/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/transforms/formatting.py -------------------------------------------------------------------------------- /superflow/datasets/transforms/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/transforms/loading.py -------------------------------------------------------------------------------- /superflow/datasets/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/datasets/transforms/transforms.py -------------------------------------------------------------------------------- /superflow/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/__init__.py -------------------------------------------------------------------------------- /superflow/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/__init__.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/dinov2_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/dinov2_vision_transformer.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/__init__.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/attention.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/block.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/dino_head.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/drop_path.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/layer_scale.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/mlp.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/patch_embed.py -------------------------------------------------------------------------------- /superflow/models/backbones/dinov2/layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/dinov2/layers/swiglu_ffn.py -------------------------------------------------------------------------------- /superflow/models/backbones/minkunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/minkunet.py -------------------------------------------------------------------------------- /superflow/models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/backbones/vit.py -------------------------------------------------------------------------------- /superflow/models/data_preprocessors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/data_preprocessors/__init__.py -------------------------------------------------------------------------------- /superflow/models/data_preprocessors/data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/data_preprocessors/data_preprocessor.py -------------------------------------------------------------------------------- /superflow/models/seg_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/seg_heads/__init__.py -------------------------------------------------------------------------------- /superflow/models/seg_heads/linear_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/seg_heads/linear_head.py -------------------------------------------------------------------------------- /superflow/models/seg_heads/upsample_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/seg_heads/upsample_head.py -------------------------------------------------------------------------------- /superflow/models/segmentors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/segmentors/__init__.py -------------------------------------------------------------------------------- /superflow/models/segmentors/voxel_segmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/segmentors/voxel_segmentor.py -------------------------------------------------------------------------------- /superflow/models/selfsup/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/selfsup/__init__.py -------------------------------------------------------------------------------- /superflow/models/selfsup/superflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/superflow/models/selfsup/superflow.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/worldbench/SuperFlow/HEAD/train.py --------------------------------------------------------------------------------