├── .gitignore ├── LICENSE ├── README.md ├── data ├── __pycache__ │ ├── basicFunction.cpython-36.pyc │ ├── basicFunction.cpython-37.pyc │ ├── dataloader.cpython-36.pyc │ └── dataloader.cpython-37.pyc ├── basicFunction.py └── dataloader.py ├── figs ├── model │ ├── BidAtentionLayer.png │ ├── model_architecture.pdf │ └── model_architecture.png └── results │ ├── GT │ ├── GT_001.png │ ├── GT_002.png │ ├── GT_003.png │ ├── GT_004.png │ └── GT_005.png │ ├── input │ ├── input_001.png │ ├── input_002.png │ ├── input_003.png │ ├── input_004.png │ └── input_005.png │ └── ours │ ├── ours_001.png │ ├── ours_002.png │ ├── ours_003.png │ ├── ours_004.png │ └── ours_005.png ├── generateMask.py ├── loss ├── InpaintingLoss.py └── __pycache__ │ └── InpaintingLoss.cpython-36.pyc ├── models ├── ActivationFunction.py ├── LBAMModel.py ├── __pycache__ │ ├── ActivationFunction.cpython-36.pyc │ ├── ActivationFunction.cpython-37.pyc │ ├── LBAMModel.cpython-36.pyc │ ├── LBAMModel.cpython-37.pyc │ ├── discriminator.cpython-36.pyc │ ├── forwardAttentionLayer.cpython-36.pyc │ ├── forwardAttentionLayer.cpython-37.pyc │ ├── reverseAttentionLayer.cpython-36.pyc │ ├── reverseAttentionLayer.cpython-37.pyc │ ├── weightInitial.cpython-36.pyc │ └── weightInitial.cpython-37.pyc ├── discriminator.py ├── forwardAttentionLayer.py ├── reverseAttentionLayer.py └── weightInitial.py ├── pytorch_ssim ├── __init__.py └── __pycache__ │ ├── __init__.cpython-36.pyc │ └── __init__.cpython-37.pyc ├── test.py ├── test_random_batch.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /data/__pycache__/basicFunction.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/__pycache__/basicFunction.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/basicFunction.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/__pycache__/basicFunction.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/dataloader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/__pycache__/dataloader.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/dataloader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/__pycache__/dataloader.cpython-37.pyc -------------------------------------------------------------------------------- /data/basicFunction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/basicFunction.py -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /figs/model/BidAtentionLayer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/model/BidAtentionLayer.png -------------------------------------------------------------------------------- /figs/model/model_architecture.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/model/model_architecture.pdf -------------------------------------------------------------------------------- /figs/model/model_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/model/model_architecture.png -------------------------------------------------------------------------------- /figs/results/GT/GT_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/GT/GT_001.png -------------------------------------------------------------------------------- /figs/results/GT/GT_002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/GT/GT_002.png -------------------------------------------------------------------------------- /figs/results/GT/GT_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/GT/GT_003.png -------------------------------------------------------------------------------- /figs/results/GT/GT_004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/GT/GT_004.png -------------------------------------------------------------------------------- /figs/results/GT/GT_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/GT/GT_005.png -------------------------------------------------------------------------------- /figs/results/input/input_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/input/input_001.png -------------------------------------------------------------------------------- /figs/results/input/input_002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/input/input_002.png -------------------------------------------------------------------------------- /figs/results/input/input_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/input/input_003.png -------------------------------------------------------------------------------- /figs/results/input/input_004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/input/input_004.png -------------------------------------------------------------------------------- /figs/results/input/input_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/input/input_005.png -------------------------------------------------------------------------------- /figs/results/ours/ours_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/ours/ours_001.png -------------------------------------------------------------------------------- /figs/results/ours/ours_002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/ours/ours_002.png -------------------------------------------------------------------------------- /figs/results/ours/ours_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/ours/ours_003.png -------------------------------------------------------------------------------- /figs/results/ours/ours_004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/ours/ours_004.png -------------------------------------------------------------------------------- /figs/results/ours/ours_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/figs/results/ours/ours_005.png -------------------------------------------------------------------------------- /generateMask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/generateMask.py -------------------------------------------------------------------------------- /loss/InpaintingLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/loss/InpaintingLoss.py -------------------------------------------------------------------------------- /loss/__pycache__/InpaintingLoss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/loss/__pycache__/InpaintingLoss.cpython-36.pyc -------------------------------------------------------------------------------- /models/ActivationFunction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/ActivationFunction.py -------------------------------------------------------------------------------- /models/LBAMModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/LBAMModel.py -------------------------------------------------------------------------------- /models/__pycache__/ActivationFunction.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/ActivationFunction.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/ActivationFunction.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/ActivationFunction.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/LBAMModel.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/LBAMModel.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/LBAMModel.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/LBAMModel.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/discriminator.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/discriminator.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/forwardAttentionLayer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/forwardAttentionLayer.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/forwardAttentionLayer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/forwardAttentionLayer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/reverseAttentionLayer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/reverseAttentionLayer.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/reverseAttentionLayer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/reverseAttentionLayer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/weightInitial.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/weightInitial.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/weightInitial.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/__pycache__/weightInitial.cpython-37.pyc -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/forwardAttentionLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/forwardAttentionLayer.py -------------------------------------------------------------------------------- /models/reverseAttentionLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/reverseAttentionLayer.py -------------------------------------------------------------------------------- /models/weightInitial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/models/weightInitial.py -------------------------------------------------------------------------------- /pytorch_ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/pytorch_ssim/__init__.py -------------------------------------------------------------------------------- /pytorch_ssim/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/pytorch_ssim/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /pytorch_ssim/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/pytorch_ssim/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/test.py -------------------------------------------------------------------------------- /test_random_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/test_random_batch.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vious/LBAM_Pytorch/HEAD/train.py --------------------------------------------------------------------------------