liudakai2 / UnsupDIS-pytorch

A pytorch implementation of UnsupervisedDeepImageStitching
GNU General Public License v3.0
112 stars 20 forks source link

Bug fixes in test.py #5

Closed tgjjj closed 2 years ago

tgjjj commented 2 years ago

Fixed a bug in test.py that caused float 0 regarded as 1. The float 0 can be a very small positive number, so using >0 to distinguish it from 1 can cause problems. This bug probably caused incorrect PSNR and SSIM values in testing and validation. Currently not sure if training is affected.