mirrors/dandere2x
1
0
Fork 0
mirror of https://github.com/akai-katto/dandere2x.git synced 2026-05-05 03:56:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

added code for mse-ssim hybrid

Browse Source
This commit is contained in:
Tyler 2020-06-27 20:00:42 -07:00
parent a011765e90
commit 340d7f1623
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dandere2x_cpp/src/Image/SSIM/SSIM.h
View File

@ -62,11 +62,14 @@ public:
double sigxy = 0;
double sigsqx = 0;
double sigsqy = 0;
double mse = 0;
for (int x = 0; x < block_size; x++) {
for (int y = 0; y < block_size; y++) {
sigsqx += pow((getLumaColor(image_A.get_color(initial_x + x, initial_y + y), color) - mx), 2);
sigsqy += pow((getLumaColor(image_B.get_color(variable_x + x, variable_y + y), color) - my), 2);
mse += pow((getLumaColor(image_A.get_color(initial_x + x, initial_y + y), color))
- (getLumaColor(image_B.get_color(initial_x + x, initial_y + y), color)), 2);
sigxy += (getLumaColor(image_A.get_color(initial_x + x, initial_y + y), color) - mx) * (getLumaColor(image_B.get_color(variable_x + x, variable_y + y), color) - my);
}
@ -85,7 +88,14 @@ public:
double denominator = (pow(mx, 2) + pow(my, 2) + c1) * (sigsqx + sigsqy + c2);
double ssim = numerator / denominator;
return ssim;
// Variables used to stabalize a weak (or zero) denominator (similar to c1 c2 in SSIM)
double d1 = pow(0.01 * (255*255), 2);
double d2 = pow(0.03 * (255*255), 2);
double inverse_mse = (1 + d1) / (mse + d2);
return ssim * inverse_mse;
}
};