├── .DS_Store ├── .gitignore ├── LICENSE ├── README.md ├── cal_flops.py ├── config.py ├── configs ├── groupmixformer.yaml ├── groupmixformer_base.yaml ├── groupmixformer_large.yaml ├── groupmixformer_miny.yaml ├── groupmixformer_small.yaml └── groupmixformer_tiny.yaml ├── data ├── __init__.py ├── build.py ├── cached_image_folder.py ├── dataset_lmdb.py ├── folder2lmdb.py ├── samplers.py └── zipreader.py ├── detection └── detection.md ├── extract_clean_ckpt.py ├── launch_scripts ├── gma_sh1.sh ├── gma_sh2.sh ├── kill_all.sh ├── run_cal_flops.sh ├── run_eval.sh ├── run_train.sh ├── start_main.sh └── test_throughput.sh ├── logger.py ├── lr_scheduler.py ├── models ├── __init__.py ├── build.py └── groupmixformer.py ├── multi_machine_start.py ├── optimizer.py ├── pics └── teaser.png ├── segmentation └── segmentation.md ├── test.py ├── train.py └── utils.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/README.md -------------------------------------------------------------------------------- /cal_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/cal_flops.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/config.py -------------------------------------------------------------------------------- /configs/groupmixformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer.yaml -------------------------------------------------------------------------------- /configs/groupmixformer_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer_base.yaml -------------------------------------------------------------------------------- /configs/groupmixformer_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer_large.yaml -------------------------------------------------------------------------------- /configs/groupmixformer_miny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer_miny.yaml -------------------------------------------------------------------------------- /configs/groupmixformer_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer_small.yaml -------------------------------------------------------------------------------- /configs/groupmixformer_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/configs/groupmixformer_tiny.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/build.py -------------------------------------------------------------------------------- /data/cached_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/cached_image_folder.py -------------------------------------------------------------------------------- /data/dataset_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/dataset_lmdb.py -------------------------------------------------------------------------------- /data/folder2lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/folder2lmdb.py -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/samplers.py -------------------------------------------------------------------------------- /data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/data/zipreader.py -------------------------------------------------------------------------------- /detection/detection.md: -------------------------------------------------------------------------------- 1 | WIP -------------------------------------------------------------------------------- /extract_clean_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/extract_clean_ckpt.py -------------------------------------------------------------------------------- /launch_scripts/gma_sh1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/gma_sh1.sh -------------------------------------------------------------------------------- /launch_scripts/gma_sh2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/gma_sh2.sh -------------------------------------------------------------------------------- /launch_scripts/kill_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/kill_all.sh -------------------------------------------------------------------------------- /launch_scripts/run_cal_flops.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/run_cal_flops.sh -------------------------------------------------------------------------------- /launch_scripts/run_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/run_eval.sh -------------------------------------------------------------------------------- /launch_scripts/run_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/run_train.sh -------------------------------------------------------------------------------- /launch_scripts/start_main.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/start_main.sh -------------------------------------------------------------------------------- /launch_scripts/test_throughput.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/launch_scripts/test_throughput.sh -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/logger.py -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/models/build.py -------------------------------------------------------------------------------- /models/groupmixformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/models/groupmixformer.py -------------------------------------------------------------------------------- /multi_machine_start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/multi_machine_start.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/optimizer.py -------------------------------------------------------------------------------- /pics/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/pics/teaser.png -------------------------------------------------------------------------------- /segmentation/segmentation.md: -------------------------------------------------------------------------------- 1 | wip -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AILab-CVC/GroupMixFormer/HEAD/utils.py --------------------------------------------------------------------------------