├── .gitignore ├── README.md ├── assets ├── afb-urr-compressed.gif ├── mask_palette.png ├── pipeline.png └── pretrain.txt ├── dataset ├── DAVIS_DS.py ├── LongVideo_DS.py ├── PreTrain_DS.py ├── YouTube_DS.py ├── __init__.py └── transforms.py ├── eval.py ├── longvideo_evaluation ├── README.md ├── dataset.py ├── davis2017 │ ├── metrics.py │ └── utils.py ├── evaluation.py ├── results.py └── score_longvideo.py ├── model ├── AFB_URR.py ├── FeatureBank.py └── __init__.py ├── myutils ├── __init__.py ├── data.py └── system.py ├── requirements.txt ├── train.py └── unify_pretrain_dataset.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | .idea/ 3 | 4 | logs/ 5 | output/ 6 | overlay/ 7 | tmp/ 8 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/README.md -------------------------------------------------------------------------------- /assets/afb-urr-compressed.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/assets/afb-urr-compressed.gif -------------------------------------------------------------------------------- /assets/mask_palette.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/assets/mask_palette.png -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /assets/pretrain.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/assets/pretrain.txt -------------------------------------------------------------------------------- /dataset/DAVIS_DS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/DAVIS_DS.py -------------------------------------------------------------------------------- /dataset/LongVideo_DS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/LongVideo_DS.py -------------------------------------------------------------------------------- /dataset/PreTrain_DS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/PreTrain_DS.py -------------------------------------------------------------------------------- /dataset/YouTube_DS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/YouTube_DS.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/eval.py -------------------------------------------------------------------------------- /longvideo_evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/README.md -------------------------------------------------------------------------------- /longvideo_evaluation/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/dataset.py -------------------------------------------------------------------------------- /longvideo_evaluation/davis2017/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/davis2017/metrics.py -------------------------------------------------------------------------------- /longvideo_evaluation/davis2017/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/davis2017/utils.py -------------------------------------------------------------------------------- /longvideo_evaluation/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/evaluation.py -------------------------------------------------------------------------------- /longvideo_evaluation/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/results.py -------------------------------------------------------------------------------- /longvideo_evaluation/score_longvideo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/longvideo_evaluation/score_longvideo.py -------------------------------------------------------------------------------- /model/AFB_URR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/model/AFB_URR.py -------------------------------------------------------------------------------- /model/FeatureBank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/model/FeatureBank.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/model/__init__.py -------------------------------------------------------------------------------- /myutils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/myutils/__init__.py -------------------------------------------------------------------------------- /myutils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/myutils/data.py -------------------------------------------------------------------------------- /myutils/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/myutils/system.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/train.py -------------------------------------------------------------------------------- /unify_pretrain_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xmlyqing00/AFB-URR/HEAD/unify_pretrain_dataset.py --------------------------------------------------------------------------------