yzxing87 / pytorch-deep-video-prior

[NeurIPS 2020] Blind Video Temporal Consistency via Deep Video Prior
118 stars 22 forks source link

IRT loss #3

Closed nicolas-dufour closed 2 years ago

nicolas-dufour commented 3 years ago

Hi, I see that the loss used for IRT in the code is different than the one in the paper. Can u pls explain why is there a difference pls?

Also, i see u divide the pixelwise difference of O_main and the net_gt by the image. What's the idea behind this? Thank you!

yzxing87 commented 2 years ago

Hi, our implementation is basically the same as our paper, except we use several epochs for warming up training.

And actually, we did the division for normalization purposes before. But I think you can safely remove the division in our current implementation.