Fix HD multi

This commit is contained in:
hzwer
2021-03-06 11:34:05 +08:00
parent 29e6801120
commit b678f7ea2e
2 changed files with 2 additions and 2 deletions

View File

@@ -147,7 +147,7 @@ python3 benchmark/MiddelBury_Other.py
python3 benchmark/HD.py
# "PSNR: 32.124"
python3 benchmark/HD_multi.py
# "PSNR: 19.92(544*1280), 30.03(720p), 26.71(1080p)"
# "PSNR: 18.89(544*1280), 28.83(720p), 24.96(1080p)"
```
## Training and Reproduction

View File

@@ -91,7 +91,7 @@ for data in name_list:
psnr = 20 * math.log10(PIXEL_MAX / math.sqrt(mse))
else:
print('Not Implement')
psnr_list.append(psnr)
psnr_list.append(psnr)
print(np.mean(psnr_list))
tot.append(np.mean(psnr_list))