├── Docker ├── Dockerfile ├── Dockerfile_detector ├── docker_commands.txt ├── os_requirements.txt └── requirements.txt ├── License.txt ├── ReViT-1.png ├── Readme.md ├── __pycache__ ├── epoch.cpython-38.pyc ├── epoch.cpython-38_BACKUP_65179.pyc ├── epoch.cpython-38_BASE_65179.pyc ├── epoch.cpython-38_LOCAL_65179.pyc ├── epoch.cpython-38_REMOTE_65179.pyc └── run.cpython-38.pyc ├── config ├── config-ReMViTv2-T.yaml ├── config-ReSwin-T.yaml └── config-ReViT-B.yaml ├── dataset ├── __pycache__ │ ├── data.cpython-38.pyc │ ├── mixup.cpython-38.pyc │ ├── rand_augment.cpython-38.pyc │ ├── random_erasing.cpython-38.pyc │ └── transform.cpython-38.pyc ├── data.py ├── mixup.py ├── rand_augment.py ├── random_erasing.py ├── transform.py └── utils.py ├── epoch.py ├── logs ├── cascade_mask_rcnn_reswin_logs.json ├── mas_rcnn_remvit_log.json ├── remvit_tiny_patch4_window_7_224.txt ├── reswin_tiny_patch4_window7_224_alpha.txt └── revit_b_patch16_224.txt ├── main.py ├── model ├── __pycache__ │ ├── attention.cpython-38.pyc │ ├── common.cpython-38.pyc │ ├── loss.cpython-38.pyc │ ├── lr_policy.cpython-38.pyc │ ├── mmvit_model.cpython-310.pyc │ ├── mmvit_model.cpython-38.pyc │ ├── mmvit_model_.cpython-38.pyc │ ├── optimizer.cpython-38.pyc │ ├── reswin.cpython-38.pyc │ ├── revit_model.cpython-38.pyc │ ├── swin.cpython-38.pyc │ └── swin_old.cpython-38.pyc ├── attention.py ├── common.py ├── loss.py ├── lr_policy.py ├── optimizer.py ├── reswin.py └── revit_model.py ├── run.py ├── utils ├── __pycache__ │ ├── distributed.cpython-38.pyc │ ├── metrics.cpython-38.pyc │ └── model_stats.cpython-38.pyc ├── distributed.py ├── metrics ├── metrics.py └── model_stats.py └── visualizer.ipynb /Docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Docker/Dockerfile -------------------------------------------------------------------------------- /Docker/Dockerfile_detector: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Docker/Dockerfile_detector -------------------------------------------------------------------------------- /Docker/docker_commands.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Docker/docker_commands.txt -------------------------------------------------------------------------------- /Docker/os_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Docker/os_requirements.txt -------------------------------------------------------------------------------- /Docker/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Docker/requirements.txt -------------------------------------------------------------------------------- /License.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/License.txt -------------------------------------------------------------------------------- /ReViT-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/ReViT-1.png -------------------------------------------------------------------------------- /Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/Readme.md -------------------------------------------------------------------------------- /__pycache__/epoch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/epoch.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/epoch.cpython-38_BACKUP_65179.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/epoch.cpython-38_BACKUP_65179.pyc -------------------------------------------------------------------------------- /__pycache__/epoch.cpython-38_BASE_65179.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/epoch.cpython-38_BASE_65179.pyc -------------------------------------------------------------------------------- /__pycache__/epoch.cpython-38_LOCAL_65179.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/epoch.cpython-38_LOCAL_65179.pyc -------------------------------------------------------------------------------- /__pycache__/epoch.cpython-38_REMOTE_65179.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/epoch.cpython-38_REMOTE_65179.pyc -------------------------------------------------------------------------------- /__pycache__/run.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/__pycache__/run.cpython-38.pyc -------------------------------------------------------------------------------- /config/config-ReMViTv2-T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/config/config-ReMViTv2-T.yaml -------------------------------------------------------------------------------- /config/config-ReSwin-T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/config/config-ReSwin-T.yaml -------------------------------------------------------------------------------- /config/config-ReViT-B.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/config/config-ReViT-B.yaml -------------------------------------------------------------------------------- /dataset/__pycache__/data.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/__pycache__/data.cpython-38.pyc -------------------------------------------------------------------------------- /dataset/__pycache__/mixup.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/__pycache__/mixup.cpython-38.pyc -------------------------------------------------------------------------------- /dataset/__pycache__/rand_augment.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/__pycache__/rand_augment.cpython-38.pyc -------------------------------------------------------------------------------- /dataset/__pycache__/random_erasing.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/__pycache__/random_erasing.cpython-38.pyc -------------------------------------------------------------------------------- /dataset/__pycache__/transform.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/__pycache__/transform.cpython-38.pyc -------------------------------------------------------------------------------- /dataset/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/data.py -------------------------------------------------------------------------------- /dataset/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/mixup.py -------------------------------------------------------------------------------- /dataset/rand_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/rand_augment.py -------------------------------------------------------------------------------- /dataset/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/random_erasing.py -------------------------------------------------------------------------------- /dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/transform.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /epoch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/epoch.py -------------------------------------------------------------------------------- /logs/cascade_mask_rcnn_reswin_logs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/logs/cascade_mask_rcnn_reswin_logs.json -------------------------------------------------------------------------------- /logs/mas_rcnn_remvit_log.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/logs/mas_rcnn_remvit_log.json -------------------------------------------------------------------------------- /logs/remvit_tiny_patch4_window_7_224.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/logs/remvit_tiny_patch4_window_7_224.txt -------------------------------------------------------------------------------- /logs/reswin_tiny_patch4_window7_224_alpha.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/logs/reswin_tiny_patch4_window7_224_alpha.txt -------------------------------------------------------------------------------- /logs/revit_b_patch16_224.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/logs/revit_b_patch16_224.txt -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/main.py -------------------------------------------------------------------------------- /model/__pycache__/attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/attention.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/loss.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/loss.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/lr_policy.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/lr_policy.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/mmvit_model.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/mmvit_model.cpython-310.pyc -------------------------------------------------------------------------------- /model/__pycache__/mmvit_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/mmvit_model.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/mmvit_model_.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/mmvit_model_.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/optimizer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/optimizer.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/reswin.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/reswin.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/revit_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/revit_model.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/swin.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/swin.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/swin_old.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/__pycache__/swin_old.cpython-38.pyc -------------------------------------------------------------------------------- /model/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/attention.py -------------------------------------------------------------------------------- /model/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/common.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/loss.py -------------------------------------------------------------------------------- /model/lr_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/lr_policy.py -------------------------------------------------------------------------------- /model/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/optimizer.py -------------------------------------------------------------------------------- /model/reswin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/reswin.py -------------------------------------------------------------------------------- /model/revit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/model/revit_model.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/run.py -------------------------------------------------------------------------------- /utils/__pycache__/distributed.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/__pycache__/distributed.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/model_stats.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/__pycache__/model_stats.cpython-38.pyc -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/metrics: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/model_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/utils/model_stats.py -------------------------------------------------------------------------------- /visualizer.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ADiko1997/ReViT/HEAD/visualizer.ipynb --------------------------------------------------------------------------------