├── LPIPS_test.py ├── README.md ├── __pycache__ ├── colorful.cpython-37.pyc ├── dataset.cpython-37.pyc ├── my_utils.cpython-37.pyc ├── utils.cpython-37.pyc └── uuu.cpython-37.pyc ├── colorful.py ├── config └── model_config.yml ├── dataset.py ├── losses ├── Adversarial.py ├── Edge.py ├── HSV.py ├── L1.py ├── Perceptual.py ├── Style.py └── __pycache__ │ ├── Adversarial.cpython-37.pyc │ ├── Edge.cpython-37.pyc │ ├── HSV.cpython-37.pyc │ ├── L1.cpython-37.pyc │ ├── Perceptual.cpython-37.pyc │ └── Style.cpython-37.pyc ├── models ├── Discriminator.py ├── RDB_layer.py ├── __pycache__ │ ├── Discriminator.cpython-37.pyc │ ├── RDB_layer.cpython-37.pyc │ ├── cswin_transformer.cpython-37.pyc │ └── model.cpython-37.pyc ├── cswin_transformer.py └── model.py ├── my_utils.py ├── requirements.txt ├── test.py ├── train.py ├── txt ├── example_test_dataset_place2.txt ├── example_train_dataset_place2.txt ├── example_train_mask_place2.txt └── example_val_dataset_place2.txt └── warmup_scheduler ├── __init__.py ├── __pycache__ ├── __init__.cpython-37.pyc └── scheduler.cpython-37.pyc ├── run.py └── scheduler.py /LPIPS_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/LPIPS_test.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/colorful.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/__pycache__/colorful.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/__pycache__/dataset.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/my_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/__pycache__/my_utils.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/uuu.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/__pycache__/uuu.cpython-37.pyc -------------------------------------------------------------------------------- /colorful.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/colorful.py -------------------------------------------------------------------------------- /config/model_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/config/model_config.yml -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/dataset.py -------------------------------------------------------------------------------- /losses/Adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/Adversarial.py -------------------------------------------------------------------------------- /losses/Edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/Edge.py -------------------------------------------------------------------------------- /losses/HSV.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/HSV.py -------------------------------------------------------------------------------- /losses/L1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/L1.py -------------------------------------------------------------------------------- /losses/Perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/Perceptual.py -------------------------------------------------------------------------------- /losses/Style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/Style.py -------------------------------------------------------------------------------- /losses/__pycache__/Adversarial.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/Adversarial.cpython-37.pyc -------------------------------------------------------------------------------- /losses/__pycache__/Edge.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/Edge.cpython-37.pyc -------------------------------------------------------------------------------- /losses/__pycache__/HSV.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/HSV.cpython-37.pyc -------------------------------------------------------------------------------- /losses/__pycache__/L1.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/L1.cpython-37.pyc -------------------------------------------------------------------------------- /losses/__pycache__/Perceptual.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/Perceptual.cpython-37.pyc -------------------------------------------------------------------------------- /losses/__pycache__/Style.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/losses/__pycache__/Style.cpython-37.pyc -------------------------------------------------------------------------------- /models/Discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/Discriminator.py -------------------------------------------------------------------------------- /models/RDB_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/RDB_layer.py -------------------------------------------------------------------------------- /models/__pycache__/Discriminator.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/__pycache__/Discriminator.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/RDB_layer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/__pycache__/RDB_layer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/cswin_transformer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/__pycache__/cswin_transformer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /models/cswin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/cswin_transformer.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/models/model.py -------------------------------------------------------------------------------- /my_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/my_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/train.py -------------------------------------------------------------------------------- /txt/example_test_dataset_place2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/txt/example_test_dataset_place2.txt -------------------------------------------------------------------------------- /txt/example_train_dataset_place2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/txt/example_train_dataset_place2.txt -------------------------------------------------------------------------------- /txt/example_train_mask_place2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/txt/example_train_mask_place2.txt -------------------------------------------------------------------------------- /txt/example_val_dataset_place2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/txt/example_val_dataset_place2.txt -------------------------------------------------------------------------------- /warmup_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/warmup_scheduler/__init__.py -------------------------------------------------------------------------------- /warmup_scheduler/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/warmup_scheduler/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /warmup_scheduler/__pycache__/scheduler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/warmup_scheduler/__pycache__/scheduler.cpython-37.pyc -------------------------------------------------------------------------------- /warmup_scheduler/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/warmup_scheduler/run.py -------------------------------------------------------------------------------- /warmup_scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bobo0303/LIGHTWEIGHT-IMAGE-INPAINTING-BY-STRIPE-WINDOW-TRANSFORMER-WITH-JOINT-ATTENTION-TO-CNN/HEAD/warmup_scheduler/scheduler.py --------------------------------------------------------------------------------