├── .idea ├── .gitignore ├── .name └── inspectionProfiles │ └── profiles_settings.xml ├── LICENSE.md ├── README.md ├── __pycache__ ├── checkpoint.cpython-36.pyc ├── checkpoint.cpython-37.pyc ├── completion_model.cpython-36.pyc ├── completion_model.cpython-37.pyc ├── completion_model.cpython-38.pyc ├── dataloader.cpython-36.pyc ├── dataloader.cpython-37.pyc ├── dataloader.cpython-38.pyc ├── loss.cpython-37.pyc ├── rginp.cpython-37.pyc ├── train.cpython-36.pyc ├── train.cpython-37.pyc ├── train.cpython-38.pyc ├── utils.cpython-36.pyc └── utils.cpython-37.pyc ├── checkpoint.py ├── completion_model.py ├── dataloader.py ├── imgs ├── architecture.jpg └── main_img.jpg ├── loss.py ├── main.py ├── 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 ├── requirements.txt ├── rginp.py ├── user_test └── user_input │ ├── image │ ├── ref │ │ └── sample.jpg │ └── src │ │ └── sample.jpg │ └── mask │ └── sample.jpg └── utils.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | 5 | -------------------------------------------------------------------------------- /.idea/.name: -------------------------------------------------------------------------------- 1 | main.py -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/checkpoint.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/checkpoint.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/checkpoint.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/checkpoint.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/completion_model.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/completion_model.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/completion_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/completion_model.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/completion_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/completion_model.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/dataloader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/dataloader.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/dataloader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/dataloader.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/rginp.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/rginp.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/train.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/train.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/train.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/train.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/train.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/train.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/checkpoint.py -------------------------------------------------------------------------------- /completion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/completion_model.py -------------------------------------------------------------------------------- /dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/dataloader.py -------------------------------------------------------------------------------- /imgs/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/imgs/architecture.jpg -------------------------------------------------------------------------------- /imgs/main_img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/imgs/main_img.jpg -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/main.py -------------------------------------------------------------------------------- /models/ActivationFunction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/ActivationFunction.py -------------------------------------------------------------------------------- /models/LBAMModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/LBAMModel.py -------------------------------------------------------------------------------- /models/__pycache__/ActivationFunction.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/ActivationFunction.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/ActivationFunction.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/ActivationFunction.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/LBAMModel.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/LBAMModel.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/LBAMModel.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/LBAMModel.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/discriminator.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/discriminator.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/forwardAttentionLayer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/forwardAttentionLayer.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/forwardAttentionLayer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/forwardAttentionLayer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/reverseAttentionLayer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/reverseAttentionLayer.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/reverseAttentionLayer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/reverseAttentionLayer.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/weightInitial.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/weightInitial.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/weightInitial.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/__pycache__/weightInitial.cpython-37.pyc -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/forwardAttentionLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/forwardAttentionLayer.py -------------------------------------------------------------------------------- /models/reverseAttentionLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/reverseAttentionLayer.py -------------------------------------------------------------------------------- /models/weightInitial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/models/weightInitial.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/requirements.txt -------------------------------------------------------------------------------- /rginp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/rginp.py -------------------------------------------------------------------------------- /user_test/user_input/image/ref/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/user_test/user_input/image/ref/sample.jpg -------------------------------------------------------------------------------- /user_test/user_input/image/src/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/user_test/user_input/image/src/sample.jpg -------------------------------------------------------------------------------- /user_test/user_input/mask/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/user_test/user_input/mask/sample.jpg -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stillrot/RGINP/HEAD/utils.py --------------------------------------------------------------------------------