flyakon / Pytorch-CloudMattingGAN

Generative Adversarial Training for Weakly Supervised Cloud Matting, in ICCV 2019
4 stars 2 forks source link

关于第二步训练 #1

Open qubewx opened 1 year ago

qubewx commented 1 year ago

请问第二部训练的时候,用到了第一步训练的结果了吗

flyakon commented 1 year ago

用到了。第二部分实际上是在第一部分的基础上引入了matting的任务。之所以分为两部分是为了在引入matting之前,可以保证GAN得到一个稳定的生成结果。

qubewx commented 1 year ago

您好,第一步训练的时候loss非常低0.000几左右,这是正常的吗?还是因为数据集太小。我看您提供的完整数据集里面的数据量也比较少,不知那个70多兆的是不是完整数据集。

flyakon commented 1 year ago

提供的就是全部的数据,现在的数据集是比较小。第一步的生成损失和判别损失都是0.000几嘛,如果你是在新数据上做的话,可能需要调整,GAN是有点玄学的。