pat-e/auto-boost-algorithm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix case of inf XPSNR scores

Browse Source 
Each plane is treated independently
This commit is contained in:
trixoniisama
2024-12-07 10:46:42 +01:00
parent dcc45cb970
commit fd459548d1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
auto-boost_2.5.py
View File

@@ -231,11 +231,11 @@ def get_xpsnr(xpsnr_txt_path):
with xpsnr_txt_path.open("r") as file:
for line in file:
match = re.search(r"XPSNR [yY]: ([0-9]+\.[0-9]+) XPSNR [uU]: ([0-9]+\.[0-9]+) XPSNR [vV]: ([0-9]+\.[0-9]+)", line)
match = re.search(r"XPSNR [yY]: ([0-9]+\.[0-9]+|inf) XPSNR [uU]: ([0-9]+\.[0-9]+|inf) XPSNR [vV]: ([0-9]+\.[0-9]+|inf)", line)
if match:
Y = float(match.group(1))
U = float(match.group(2))
V = float(match.group(3))
Y = float(match.group(1)) if match.group(1) != 'inf' else 100.0
U = float(match.group(2)) if match.group(2) != 'inf' else 100.0
V = float(match.group(3)) if match.group(3) != 'inf' else 100.0
W = (4 * Y + U + V) / 6
sum_weighted += W