pavancm / CONTRIQUE

Official implementation for "Image Quality Assessment using Contrastive Learning"

Repository from Github https://github.compavancm/CONTRIQUERepository from Github https://github.compavancm/CONTRIQUE

岭回归

tonyhuang33 opened this issue · comments

我想问一下你们是怎么实现,分批次图像质量评估数据集来训练岭回归网络,最终得到一个训练模型的,我只能复现一个批次一个训练模型,怎么把多个批次的模型整合成一个训练模型呢?你们有相关代码提供吗?

In my current implementation the ridge regression is trained on the entire dataset in one shot (not in batches). Note that I don't combine multiple IQA datasets for training ridge regressor, only a single IQA dataset is used. I haven't tried batch training for ridge regression but it should be possible to do that in PyTorch.

thanks for replying , i will try to concatenated these features into one feature file.

i have a question to ask. When I use your regression model for IQA testing, how can I determine if the test set I am using contains your training data?

The data we used to train the model is from different public datasets and we have listed the names of those datasets in the paper. During testing please make sure there is no overlap of images from these datasets.