Fix some small bugs with SSIM.

This commit is contained in:
Pavel Krajcevski 2013-10-11 13:05:59 -04:00
parent 8d37d6eee5
commit 2d8c291107
2 changed files with 7 additions and 3 deletions

View file

@ -106,7 +106,7 @@ namespace FasTC {
}
double ComputePSNR(Image<PixelType> *other);
double ComputeMSSIM(Image<PixelType> *other) const;
double ComputeMSSIM(Image<PixelType> *other);
// Function to allow derived classes to populate the pixel array.
// This may involve decompressing a compressed image or otherwise

View file

@ -222,6 +222,7 @@ template<typename PixelType>
static Image<PixelType> FilterValid(const Image<PixelType> &img, uint32 size, double sigma) {
assert(size % 2);
Image<IPixel> gaussian(size, size);
GenerateGaussianKernel(gaussian, size, sigma);
Image<PixelType> tmp = img;
tmp.Filter(gaussian);
@ -237,7 +238,7 @@ static Image<PixelType> FilterValid(const Image<PixelType> &img, uint32 size, do
}
template<typename PixelType>
double Image<PixelType>::ComputeMSSIM(Image<PixelType> *other) const {
double Image<PixelType>::ComputeSSIM(Image<PixelType> *other) {
if(!other) {
return -1.0;
}
@ -247,6 +248,9 @@ double Image<PixelType>::ComputeMSSIM(Image<PixelType> *other) const {
return -1.0;
}
ComputePixels();
other->ComputePixels();
double C1 = (0.01 * 255.0 * 0.01 * 255.0);
double C2 = (0.03 * 255.0 * 0.03 * 255.0);