summaryrefslogtreecommitdiff
path: root/text_recognizer/data
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-27 22:27:24 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-27 22:27:24 +0200
commit9a8044f4a3826a119416665741b709cd686fca87 (patch)
treee339593bb4e3858fa9379d14752dc52bf5949825 /text_recognizer/data
parentae8bfa62f0e02bd70c27bc1e71697249a5a79e7e (diff)
Remove Barlow Twins
Diffstat (limited to 'text_recognizer/data')
-rw-r--r--text_recognizer/data/transforms/barlow.py19
1 files changed, 0 insertions, 19 deletions
diff --git a/text_recognizer/data/transforms/barlow.py b/text_recognizer/data/transforms/barlow.py
deleted file mode 100644
index 78683cb..0000000
--- a/text_recognizer/data/transforms/barlow.py
+++ /dev/null
@@ -1,19 +0,0 @@
-"""Augmentations for training Barlow Twins."""
-from omegaconf.dictconfig import DictConfig
-from torch import Tensor
-
-from text_recognizer.data.transforms.load_transform import load_transform
-
-
-class BarlowTransform:
- """Applies two different transforms to input data."""
-
- def __init__(self, prim: DictConfig, bis: DictConfig) -> None:
- self.prim = load_transform(prim)
- self.bis = load_transform(bis)
-
- def __call__(self, data: Tensor) -> Tensor:
- """Applies two different augmentation on the input."""
- x_prim = self.prim(data)
- x_bis = self.bis(data)
- return x_prim, x_bis