yuzu/FasTC
1
0
Fork 0
mirror of https://github.com/yuzu-emu/FasTC.git synced 2025-01-09 01:55:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

SSIM measurement is usually the smallest SSIm index, so report that instead of MSSIM, but if the user wants SSIM we can do that, too.

Browse source
This commit is contained in:
Pavel Krajcevski 2013-10-11 13:06:29 -04:00
parent 2d8c291107
commit 855683983c
3 changed files with 23 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Base/include/Image.h
View file

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

22
Base/src/Image.cpp
View file

@ -238,7 +238,7 @@ static Image<PixelType> FilterValid(const Image<PixelType> &img, uint32 size, do
} }
template<typename PixelType> template<typename PixelType>
double Image<PixelType>::ComputeSSIM(Image<PixelType> *other) { double Image<PixelType>::ComputeSSIM(Image<PixelType> *other, double *mssim) {
if(!other) { if(!other) {
return -1.0; return -1.0;
} }
@ -361,7 +361,11 @@ double Image<PixelType>::ComputeSSIM(Image<PixelType> *other) {
} }
} }
double mssim = 0.0; double minSSIM = 1.0;
if(mssim) {
*mssim = 0.0;
}
for(uint32 j = 0; j < h; j++) { for(uint32 j = 0; j < h; j++) {
for(uint32 i = 0; i < w; i++) { for(uint32 i = 0; i < w; i++) {
double m1sq = static_cast<float>(mu1_sq(i, j)); double m1sq = static_cast<float>(mu1_sq(i, j));
@ -372,13 +376,23 @@ double Image<PixelType>::ComputeSSIM(Image<PixelType> *other) {
double s2sq = static_cast<float>(sigma2_sq(i, j)); double s2sq = static_cast<float>(sigma2_sq(i, j));
double s1s2 = static_cast<float>(sigma12(i, j)); double s1s2 = static_cast<float>(sigma12(i, j));
mssim += double ssim =
((2.0 * m1m2 + C1) * (2.0 * s1s2 + C2)) / ((2.0 * m1m2 + C1) * (2.0 * s1s2 + C2)) /
((m1sq + m2sq + C1) * (s1sq + s2sq + C2)); ((m1sq + m2sq + C1) * (s1sq + s2sq + C2));
if(mssim) {
*mssim += ssim;
}
minSSIM = ::std::min(ssim, minSSIM);
} }
} }
return mssim / static_cast<double>(w * h); if(mssim) {
*mssim /= static_cast<double>(w * h);
}
return minSSIM;
} }
// !FIXME! These won't work for non-RGBA8 data. // !FIXME! These won't work for non-RGBA8 data.

5
Base/test/TestImage.cpp
View file

@ -164,5 +164,8 @@ TEST(Image, ComputeMSSIM) {
} }
} }
EXPECT_EQ(img.ComputeMSSIM(&img), 1.0); double MSSIM;
double SSIM = img.ComputeSSIM(&img, &MSSIM);
EXPECT_EQ(SSIM, 1.0);
EXPECT_EQ(MSSIM, 1.0);
} }