├── .gitignore ├── README.md ├── demo ├── img │ ├── I16_01_1.bmp │ ├── I22_01_1.bmp │ ├── i15_10_5.bmp │ ├── i16_03_3.bmp │ ├── i16_23_5.bmp │ ├── i22_21_4.bmp │ ├── img001.bmp │ └── img_info.txt └── predict.py ├── src ├── datasets │ ├── __init__.py │ ├── clive_dataloader.py │ ├── iqa_dataloader.py │ └── rank_dataloader.py ├── loss │ ├── EMD_loss.py │ ├── rank_loss.py │ └── reg_loss.py ├── metrics │ └── srocc.py ├── net │ ├── VGG16_model.py │ └── model.py └── utils │ ├── checkpoint.py │ └── logger.py └── tools ├── evaluate.py ├── train_clive.py ├── train_iqa.py └── train_rank.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/README.md -------------------------------------------------------------------------------- /demo/img/I16_01_1.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/I16_01_1.bmp -------------------------------------------------------------------------------- /demo/img/I22_01_1.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/I22_01_1.bmp -------------------------------------------------------------------------------- /demo/img/i15_10_5.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/i15_10_5.bmp -------------------------------------------------------------------------------- /demo/img/i16_03_3.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/i16_03_3.bmp -------------------------------------------------------------------------------- /demo/img/i16_23_5.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/i16_23_5.bmp -------------------------------------------------------------------------------- /demo/img/i22_21_4.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/i22_21_4.bmp -------------------------------------------------------------------------------- /demo/img/img001.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/img001.bmp -------------------------------------------------------------------------------- /demo/img/img_info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/img/img_info.txt -------------------------------------------------------------------------------- /demo/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/demo/predict.py -------------------------------------------------------------------------------- /src/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/datasets/__init__.py -------------------------------------------------------------------------------- /src/datasets/clive_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/datasets/clive_dataloader.py -------------------------------------------------------------------------------- /src/datasets/iqa_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/datasets/iqa_dataloader.py -------------------------------------------------------------------------------- /src/datasets/rank_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/datasets/rank_dataloader.py -------------------------------------------------------------------------------- /src/loss/EMD_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/loss/EMD_loss.py -------------------------------------------------------------------------------- /src/loss/rank_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/loss/rank_loss.py -------------------------------------------------------------------------------- /src/loss/reg_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/loss/reg_loss.py -------------------------------------------------------------------------------- /src/metrics/srocc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/metrics/srocc.py -------------------------------------------------------------------------------- /src/net/VGG16_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/net/VGG16_model.py -------------------------------------------------------------------------------- /src/net/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/net/model.py -------------------------------------------------------------------------------- /src/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/utils/checkpoint.py -------------------------------------------------------------------------------- /src/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/src/utils/logger.py -------------------------------------------------------------------------------- /tools/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/tools/evaluate.py -------------------------------------------------------------------------------- /tools/train_clive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/tools/train_clive.py -------------------------------------------------------------------------------- /tools/train_iqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/tools/train_iqa.py -------------------------------------------------------------------------------- /tools/train_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zee0518/TF_RankIQA/HEAD/tools/train_rank.py --------------------------------------------------------------------------------