megvii-model / HINet

Other
367 stars 62 forks source link

关于预训练模型 #44

Open ldlshizhu opened 2 years ago

ldlshizhu commented 2 years ago

Hi,您好,非常感激您所做的工作, 请问您的预训练模型是如何得到的呢,我看到好像是用HINet在不同的小数据集(比如Gopro/SSID等)上预训练得到的(?),而不是用类似vgg这样的通用模型在ImageNet上预训练得到的。 如果我想做一些调整比如更改您的结构,是不是也要用和您同样的方式进行预训练呢

mayorx commented 2 years ago

ldlshizhu 您好,

我们是在GoPro/SIDD 上从头开始训练得到的预训练模型, 目的是为了和当前的一些(同样是从头训练的模型)进行公平的比较。 您可以随意修改网络结构,然后在您的任务上直接训练即可。 感谢!

ldlshizhu commented 2 years ago

ldlshizhu 您好,

我们是在GoPro/SIDD 上从头开始训练得到的预训练模型, 目的是为了和当前的一些(同样是从头训练的模型)进行公平的比较。 您可以随意修改网络结构,然后在您的任务上直接训练即可。 感谢!

谢谢您!祝您生活愉快!