add license

Author: lucidtronix

ml4h/models/diffusion_blocks.py

@@ -781,7 +781,9 @@ def metrics(self):
     def denormalize(self, images):
         # convert the pixel values back to 0-1 range
         # images = images - tf.math.reduce_mean(images) + images * tf.math.reduce_std(images)
-        images = self.normalizer.mean + images * self.normalizer.variance ** 0.5
+        std = tf.sqrt(self.normalizer.variance)
+        images = images * std + self.normalizer.mean
+        #images = self.normalizer.mean + images * self.normalizer.variance ** 0.5
         # print(f'images max min {images}')
         return images  # tf.clip_by_value(images, 0.0, 1.0)