From f2cd16f340aa11afadb8fa90c29f85ca1b75a600 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Mon, 16 Nov 2020 20:26:32 +0100
Subject: Added a whitening transform.
---
src/text_recognizer/datasets/transforms.py | 8 ++++++++
1 file changed, 8 insertions(+)
(limited to 'src/text_recognizer/datasets')
diff --git a/src/text_recognizer/datasets/transforms.py b/src/text_recognizer/datasets/transforms.py
index 1105f23..d1ca127 100644
--- a/src/text_recognizer/datasets/transforms.py
+++ b/src/text_recognizer/datasets/transforms.py
@@ -64,3 +64,11 @@ class AddTokens:
target = torch.cat([sos, target], dim=0)
return target
+
+
+class Whitening:
+ """Whitening of Tensor, i.e. set mean to zero and std to one."""
+
+ def __call__(self, x: Tensor) -> Tensor:
+ """Apply the whitening."""
+ return (x - x.mean()) / x.std()
--
cgit v1.2.3-70-g09d2