search

cecret3350 / DEA-Net

[IEEE TIP 2024] DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attention
217 stars 10 forks source link

作者您好，为何用code/train.py 训练后生成的是.pk文件而不是.pth文件，然后使用eval.py运行的时候在加载网络的时候报错 #14

Closed xuexisuanfa closed 6 months ago

xuexisuanfa commented 9 months ago

作者您好，为何用code/train.py 训练后生成的是.pk文件而不是.pth文件，然后使用eval.py运行的时候在加载网络的时候报错

cecret3350 commented 9 months ago

作者您好，为何用code/train.py 训练后生成的是.pk文件而不是.pth文件，然后使用eval.py运行的时候在加载网络的时候报错

您好，关于pk和pth后缀，可以在保存权重或者读取权重的时候更改一下后缀即可。 使用eval.py加载train.py训练后的模型报错，是因为train.py训练后的模型还没有经过重参数化。可以修改eval.py，将定义的网络修改成训练时的网络。