sunshangquan / Histoformer

[ECCV 2024] Histoformer: Restoring Images in Adverse Weather Conditions via Histogram Transformer
126 stars 10 forks source link

数据训练的问题 #14

Open gejutaixiao22 opened 4 hours ago

gejutaixiao22 commented 4 hours ago

您好,我想询问一下模型中有关数据训练的部分。我注意到文中有三个任务,去雪,去雨雾以及去雨滴。我想询问一下这个程序是如何进行训练的。我注意到yml文件中是训练了一个数据集,并对每个数据集进行验证。请问真实训练时候是这样么。如果是的话那对哪个数据集进行训练呢。如果不是这样,那是分别对三个数据集进行训练并分别测试么,因为我也注意到三个数据集都有各自的训练集和测试集,如果是这样,RainDrop数据集我对test_a进行测试的。这是否正确呢。谢谢。

sunshangquan commented 3 hours ago

您好,这个任务仿照的是transweather,把三个数据集的训练集混合起来一起训练,然后使用训练出来的一个模型分别对三个测试集测试。RainDrop数据集是对test_a进行测试

gejutaixiao22 commented 1 hour ago

那混在一起训练的是在您给的链接中Snow100K数据集么,我看这个数据集有个Allweather的txt文件,总共是18069张。

sunshangquan commented 1 hour ago

训练集是三个数据集训练集的合集,使用的是TransWeatherAll-in-One (CVPR 2020提供的,请见下载链接