Skip to content
Snippets Groups Projects

Make normalization optional

Merged Manon Blanco requested to merge optional-normalization into main
All threads resolved!
1 file
+ 3
0
Compare changes
  • Side-by-side
  • Inline
@@ -125,6 +125,9 @@ class DAN:
image = read_image(path)
preprocessed_image = self.preprocessing_transforms(image)
if self.mean is None and self.std is None:
return preprocessed_image, preprocessed_image
size = preprocessed_image.shape
normalized_image = torch.zeros(size)
Loading