SSIM / MS-SSIM implementation about adversarial_video_generation HOT 4 OPEN

Good catch. Fixed!

from adversarial_video_generation.

Hey @bsautermeister, thanks for the link! I'll check that out and integrate it when I have a chance. Or, if you'd like to add it in yourself and submit a pull request, I'd be happy to accept.

from adversarial_video_generation.

I will, as soon as I find time for that.

By the way: I'm not sure, but after comparing your PSNR implementation with it's formula in the paper and the PSNR Wikipedia-Definition for colored images:

For color images with three RGB values per pixel, the definition of PSNR is the same except the MSE is the sum over all squared value differences divided by image size and by three.

I think that you miss the by three. As you are using the formula for colored images and take the sum over all 3 dimension (tf.reduce_sum(...,[1, 2, 3]), I think you should consider the 3rd dimension for your num_pixels as well:

num_pixels = tf.to_float(shape[1] * shape[2] * shape[3])

Or am I wrong? :)

EDIT

Probably the same thing for sharpness-differences error as well...

from adversarial_video_generation.

I know this is very very late but I wanted to get the SSIM @dyelax or @bsautermeister, did one of you got the chance to integrate it with the code, or Is there any other way to calculate the SSIM for this paper like for example, compare_ssim or tf.image.ssim, even can we calculate the psnr with tf.image.psnr..... I would really appreciate a reply. Thank you for this helpful code @dyelax

from adversarial_video_generation.