├── .gitignore ├── README.md ├── ckpt └── .gitkeep ├── config.py ├── core ├── __init__.py ├── build.py ├── test.py ├── test_woGT.py ├── train.py └── val.py ├── datasets ├── json_files │ ├── BSD100.json │ ├── DIV2K.json │ ├── DIV2K_val.json │ ├── Manga109.json │ ├── Set14.json │ ├── Set5.json │ └── Urban100.json ├── make_json.py ├── prepare_DIV2K_subimages.py └── prepare_TEST_datasets.m ├── install.sh ├── losses ├── __init__.py ├── losses.py └── metrics.py ├── models ├── IGNN.py ├── VGG19.py ├── __init__.py ├── common.py ├── graph_agg.py ├── lib │ ├── .gitignore │ ├── matmul.cpp │ ├── matmul1_bwd_kernel.cu │ ├── matmul1_kernel.cu │ ├── ops.py │ └── setup.py └── submodules.py ├── requirements.txt ├── runner.py └── utils ├── __init__.py ├── data_loaders.py ├── data_transforms.py ├── imresize.py └── network_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/config.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/core/build.py -------------------------------------------------------------------------------- /core/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/core/test.py -------------------------------------------------------------------------------- /core/test_woGT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/core/test_woGT.py -------------------------------------------------------------------------------- /core/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/core/train.py -------------------------------------------------------------------------------- /core/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/core/val.py -------------------------------------------------------------------------------- /datasets/json_files/BSD100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/BSD100.json -------------------------------------------------------------------------------- /datasets/json_files/DIV2K.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/DIV2K.json -------------------------------------------------------------------------------- /datasets/json_files/DIV2K_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/DIV2K_val.json -------------------------------------------------------------------------------- /datasets/json_files/Manga109.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/Manga109.json -------------------------------------------------------------------------------- /datasets/json_files/Set14.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/Set14.json -------------------------------------------------------------------------------- /datasets/json_files/Set5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/Set5.json -------------------------------------------------------------------------------- /datasets/json_files/Urban100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/json_files/Urban100.json -------------------------------------------------------------------------------- /datasets/make_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/make_json.py -------------------------------------------------------------------------------- /datasets/prepare_DIV2K_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/prepare_DIV2K_subimages.py -------------------------------------------------------------------------------- /datasets/prepare_TEST_datasets.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/datasets/prepare_TEST_datasets.m -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/install.sh -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/losses/losses.py -------------------------------------------------------------------------------- /losses/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/losses/metrics.py -------------------------------------------------------------------------------- /models/IGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/IGNN.py -------------------------------------------------------------------------------- /models/VGG19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/VGG19.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/common.py -------------------------------------------------------------------------------- /models/graph_agg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/graph_agg.py -------------------------------------------------------------------------------- /models/lib/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/.gitignore -------------------------------------------------------------------------------- /models/lib/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/matmul.cpp -------------------------------------------------------------------------------- /models/lib/matmul1_bwd_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/matmul1_bwd_kernel.cu -------------------------------------------------------------------------------- /models/lib/matmul1_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/matmul1_kernel.cu -------------------------------------------------------------------------------- /models/lib/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/ops.py -------------------------------------------------------------------------------- /models/lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/lib/setup.py -------------------------------------------------------------------------------- /models/submodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/models/submodules.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/runner.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/utils/data_loaders.py -------------------------------------------------------------------------------- /utils/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/utils/data_transforms.py -------------------------------------------------------------------------------- /utils/imresize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/utils/imresize.py -------------------------------------------------------------------------------- /utils/network_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sczhou/IGNN/HEAD/utils/network_utils.py --------------------------------------------------------------------------------