search

geekyutao / GraphIQA

GraphIQA: Learning Distortion Graph Representations for Blind Image Quality Assessment
Apache License 2.0
42 stars 8 forks source link

About training #1

Closed zmm96 closed 3 years ago

zmm96 commented 3 years ago

In pretraining, The dataset you used is Kondid-10k or Kondis-700k? you mean pretraining GraphIQA model from scratch on kadid10k dataset but the code seems only Kondis-700k?

sunsm-2020 commented 3 years ago

The model can be pre-trained on both kadid-10k and kadis-700k from scratch by running 'python pretrain.py --dataset kadid-P' and 'python pretrain.py --dataset kadis-P' correspondingly.