pavancm / CONTRIQUE

Official implementation for "Image Quality Assessment using Contrastive Learning"
125 stars 12 forks source link

岭回归 #20

Closed tonyhuang33 closed 6 months ago

tonyhuang33 commented 1 year ago

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

pavancm commented 1 year ago

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.

tonyhuang33 commented 1 year ago

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

tonyhuang33 commented 1 year ago

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?

pavancm commented 1 year ago

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.

tonyhuang33 commented 1 year ago

这是来自QQ邮箱的假期自动回复邮件。   您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。

tonyhuang33 commented 6 months ago

这是来自QQ邮箱的假期自动回复邮件。   您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。