caiyuanhao1998 / MST-plus-plus

"MST++: Multi-stage Spectral-wise Transformer for Efficient Spectral Reconstruction" (CVPRW 2022) & (Winner of NTIRE 2022 Spectral Recovery Challenge) and a toolbox for spectral reconstruction

Home Page:https://arxiv.org/abs/2204.07908

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

训练自己的数据集

dongruoyu1 opened this issue · comments

8A341447B610373F24AFC2D426950746
您好,我将31通道改为28通道,训练自己的数据集,结果却出现了train_losses.avg非常奇怪,甚至等于nan的情况,您有遇到过这种问题吗?

emmm,没有,建议你先对自己的数据集来一次 normalization,再训练

emmm,没有,建议你先对自己的数据集来一次 normalization,再训练

数据集我用的您在mst文章中的30张cave_512,我找到了对应的RGB图像进行训练,也按照您在mst中的步骤,将其归一化了
image
然而误差太奇怪了