├── .gitignore ├── LICENSE ├── README.md ├── datasets ├── __init__.py ├── collate_fn.py ├── segmim.py ├── texterase.py └── transforms.py ├── engine ├── __init__.py ├── train.py ├── train_segmim.py └── val.py ├── eval ├── __init__.py ├── dataloader.py ├── evaluation.py └── gauss.py ├── figures ├── segmim.svg └── viteraser.svg ├── main.py ├── main_segmim.py ├── models ├── __init__.py ├── block.py ├── decoder │ ├── __init__.py │ └── swinv2_decoder.py ├── discriminator.py ├── encoder │ ├── __init__.py │ ├── swin_transformer_v2.py │ └── swinv2_encoder.py ├── seg_head.py ├── segmim.py ├── vgg16.py └── viteraser.py ├── optim ├── __init__.py ├── loss │ ├── __init__.py │ ├── segmim.py │ └── text_removal.py ├── lr_scheduler │ ├── __init__.py │ ├── cosine.py │ ├── linear_decay.py │ └── multistep.py └── optimizer │ ├── __init__.py │ ├── segmim.py │ └── viteraser.py ├── requirements.txt ├── scripts ├── segmim │ ├── viteraser-base-encoder-finetune.sh │ ├── viteraser-base-segmim.sh │ ├── viteraser-small-encoder-finetune.sh │ ├── viteraser-small-segmim.sh │ ├── viteraser-tiny-encoder-finetune.sh │ └── viteraser-tiny-segmim.sh ├── viteraser-training-withsegmim │ ├── viteraser-base-train-withsegmim.sh │ ├── viteraser-small-train-withsegmim.sh │ └── viteraser-tiny-train-withsegmim.sh └── viteraser-training-wosegmim │ ├── viteraser-base-train.sh │ ├── viteraser-small-train.sh │ └── viteraser-tiny-train.sh ├── tools └── generate_mask.py └── utils ├── __init__.py ├── checkpointer.py ├── dist.py ├── logger.py ├── misc.py ├── parser.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | output 3 | pretrained 4 | debug 5 | /data -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/collate_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/datasets/collate_fn.py -------------------------------------------------------------------------------- /datasets/segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/datasets/segmim.py -------------------------------------------------------------------------------- /datasets/texterase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/datasets/texterase.py -------------------------------------------------------------------------------- /datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/datasets/transforms.py -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engine/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/engine/train.py -------------------------------------------------------------------------------- /engine/train_segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/engine/train_segmim.py -------------------------------------------------------------------------------- /engine/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/engine/val.py -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/eval/dataloader.py -------------------------------------------------------------------------------- /eval/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/eval/evaluation.py -------------------------------------------------------------------------------- /eval/gauss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/eval/gauss.py -------------------------------------------------------------------------------- /figures/segmim.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/figures/segmim.svg -------------------------------------------------------------------------------- /figures/viteraser.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/figures/viteraser.svg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/main.py -------------------------------------------------------------------------------- /main_segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/main_segmim.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/block.py -------------------------------------------------------------------------------- /models/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/decoder/__init__.py -------------------------------------------------------------------------------- /models/decoder/swinv2_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/decoder/swinv2_decoder.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/encoder/__init__.py -------------------------------------------------------------------------------- /models/encoder/swin_transformer_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/encoder/swin_transformer_v2.py -------------------------------------------------------------------------------- /models/encoder/swinv2_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/encoder/swinv2_encoder.py -------------------------------------------------------------------------------- /models/seg_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/seg_head.py -------------------------------------------------------------------------------- /models/segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/segmim.py -------------------------------------------------------------------------------- /models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/vgg16.py -------------------------------------------------------------------------------- /models/viteraser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/models/viteraser.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/loss/__init__.py -------------------------------------------------------------------------------- /optim/loss/segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/loss/segmim.py -------------------------------------------------------------------------------- /optim/loss/text_removal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/loss/text_removal.py -------------------------------------------------------------------------------- /optim/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/lr_scheduler/__init__.py -------------------------------------------------------------------------------- /optim/lr_scheduler/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/lr_scheduler/cosine.py -------------------------------------------------------------------------------- /optim/lr_scheduler/linear_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/lr_scheduler/linear_decay.py -------------------------------------------------------------------------------- /optim/lr_scheduler/multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/lr_scheduler/multistep.py -------------------------------------------------------------------------------- /optim/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/optimizer/__init__.py -------------------------------------------------------------------------------- /optim/optimizer/segmim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/optimizer/segmim.py -------------------------------------------------------------------------------- /optim/optimizer/viteraser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/optim/optimizer/viteraser.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/segmim/viteraser-base-encoder-finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-base-encoder-finetune.sh -------------------------------------------------------------------------------- /scripts/segmim/viteraser-base-segmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-base-segmim.sh -------------------------------------------------------------------------------- /scripts/segmim/viteraser-small-encoder-finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-small-encoder-finetune.sh -------------------------------------------------------------------------------- /scripts/segmim/viteraser-small-segmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-small-segmim.sh -------------------------------------------------------------------------------- /scripts/segmim/viteraser-tiny-encoder-finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-tiny-encoder-finetune.sh -------------------------------------------------------------------------------- /scripts/segmim/viteraser-tiny-segmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/segmim/viteraser-tiny-segmim.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-withsegmim/viteraser-base-train-withsegmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-withsegmim/viteraser-base-train-withsegmim.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-withsegmim/viteraser-small-train-withsegmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-withsegmim/viteraser-small-train-withsegmim.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-withsegmim/viteraser-tiny-train-withsegmim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-withsegmim/viteraser-tiny-train-withsegmim.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-wosegmim/viteraser-base-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-wosegmim/viteraser-base-train.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-wosegmim/viteraser-small-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-wosegmim/viteraser-small-train.sh -------------------------------------------------------------------------------- /scripts/viteraser-training-wosegmim/viteraser-tiny-train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/scripts/viteraser-training-wosegmim/viteraser-tiny-train.sh -------------------------------------------------------------------------------- /tools/generate_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/tools/generate_mask.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /utils/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/checkpointer.py -------------------------------------------------------------------------------- /utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/dist.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shannanyinxiang/ViTEraser/HEAD/utils/visualize.py --------------------------------------------------------------------------------