├── .gitignore ├── .idea └── .gitignore ├── README.md ├── data ├── __init__.py ├── datasets │ ├── __init__.py │ ├── csiq.py │ ├── kadid10k.py │ ├── kadis700k.py │ ├── koniq10k.py │ ├── live.py │ ├── pieapp_dataset.py │ ├── pipal.py │ └── tid.py ├── patch_datasets.py ├── patch_sampling.py └── utils.py ├── figures ├── patch_sampling-multiscale.png ├── patch_sampling-singlescale.png ├── patch_sampling.png ├── vtamiq_diagram_compact.png └── vtamiq_vit_diagram_compact.png ├── modules ├── Attention │ └── deepgaze2 │ │ ├── Create_Centerbias.ipynb │ │ └── centerbias.npy ├── RCAN │ └── channel_attention.py ├── VisionTransformer │ ├── __init__.py │ ├── backbone.py │ └── transformer.py ├── utils.py └── vtamiq │ ├── common.py │ └── vtamiq.py ├── run_custom.py ├── run_multi.py ├── train.py ├── train_config.py └── utils ├── __init__.py ├── image_processing ├── color_spaces.py └── image_tools.py ├── logging ├── __init__.py └── logger.py ├── matlab └── matlab_wrapper.py └── misc ├── accumulators.py ├── correlations.py ├── iterators.py ├── miscelaneous.py ├── summary_writer.py ├── temporary_numpy_seed.py └── timer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/datasets/csiq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/csiq.py -------------------------------------------------------------------------------- /data/datasets/kadid10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/kadid10k.py -------------------------------------------------------------------------------- /data/datasets/kadis700k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/kadis700k.py -------------------------------------------------------------------------------- /data/datasets/koniq10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/koniq10k.py -------------------------------------------------------------------------------- /data/datasets/live.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/live.py -------------------------------------------------------------------------------- /data/datasets/pieapp_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/pieapp_dataset.py -------------------------------------------------------------------------------- /data/datasets/pipal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/pipal.py -------------------------------------------------------------------------------- /data/datasets/tid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/datasets/tid.py -------------------------------------------------------------------------------- /data/patch_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/patch_datasets.py -------------------------------------------------------------------------------- /data/patch_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/patch_sampling.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/data/utils.py -------------------------------------------------------------------------------- /figures/patch_sampling-multiscale.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/figures/patch_sampling-multiscale.png -------------------------------------------------------------------------------- /figures/patch_sampling-singlescale.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/figures/patch_sampling-singlescale.png -------------------------------------------------------------------------------- /figures/patch_sampling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/figures/patch_sampling.png -------------------------------------------------------------------------------- /figures/vtamiq_diagram_compact.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/figures/vtamiq_diagram_compact.png -------------------------------------------------------------------------------- /figures/vtamiq_vit_diagram_compact.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/figures/vtamiq_vit_diagram_compact.png -------------------------------------------------------------------------------- /modules/Attention/deepgaze2/Create_Centerbias.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/Attention/deepgaze2/Create_Centerbias.ipynb -------------------------------------------------------------------------------- /modules/Attention/deepgaze2/centerbias.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/Attention/deepgaze2/centerbias.npy -------------------------------------------------------------------------------- /modules/RCAN/channel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/RCAN/channel_attention.py -------------------------------------------------------------------------------- /modules/VisionTransformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/VisionTransformer/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/VisionTransformer/backbone.py -------------------------------------------------------------------------------- /modules/VisionTransformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/VisionTransformer/transformer.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/utils.py -------------------------------------------------------------------------------- /modules/vtamiq/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/vtamiq/common.py -------------------------------------------------------------------------------- /modules/vtamiq/vtamiq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/modules/vtamiq/vtamiq.py -------------------------------------------------------------------------------- /run_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/run_custom.py -------------------------------------------------------------------------------- /run_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/run_multi.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/train.py -------------------------------------------------------------------------------- /train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/train_config.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/image_processing/color_spaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/image_processing/color_spaces.py -------------------------------------------------------------------------------- /utils/image_processing/image_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/image_processing/image_tools.py -------------------------------------------------------------------------------- /utils/logging/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/logging/__init__.py -------------------------------------------------------------------------------- /utils/logging/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/logging/logger.py -------------------------------------------------------------------------------- /utils/matlab/matlab_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/matlab/matlab_wrapper.py -------------------------------------------------------------------------------- /utils/misc/accumulators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/accumulators.py -------------------------------------------------------------------------------- /utils/misc/correlations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/correlations.py -------------------------------------------------------------------------------- /utils/misc/iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/iterators.py -------------------------------------------------------------------------------- /utils/misc/miscelaneous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/miscelaneous.py -------------------------------------------------------------------------------- /utils/misc/summary_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/summary_writer.py -------------------------------------------------------------------------------- /utils/misc/temporary_numpy_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/temporary_numpy_seed.py -------------------------------------------------------------------------------- /utils/misc/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ch-andrei/VTAMIQ/HEAD/utils/misc/timer.py --------------------------------------------------------------------------------