psychopa4 / PFNL

Progressive Fusion Video Super-Resolution Network via Exploiting Non-Local Spatio-Temporal Correlations
MIT License
138 stars 21 forks source link

关于VESPCN模型的训练参数 #8

Open pbdahzou opened 4 years ago

pbdahzou commented 4 years ago

psychopa4 你好,非常感谢你开源了PFNL算法以及其他SOTA算法的实现。 目前我在用你的代码重新训练VESPCN模型,使用你提供的数据和代码里默认的参数,但经过5万多次迭代训练后,PSNR还是只有25dB左右。能否告诉一下你训练时的参数哈? 谢谢!

Easoncyx commented 4 years ago

我也有同样的问题,请问您解决了吗?

pbdahzou commented 4 years ago

我也有同样的问题,请问您解决了吗? 还没解决哈。 后面忙别的去了。 之前尝试时是直接用源码训的,但源码里没有先训练flow部分,如果先训好flow,再训SR,最后联合训练可能会好点。如有进展也请告知哈,谢谢。

Easoncyx commented 4 years ago

原因是训练数据和验证数据预处理方式不一样,训练数据是三次插值,验证数据是高斯降采样。自己生成一致的数据就好了