diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:27:24 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:27:24 +0200 |
commit | 9a8044f4a3826a119416665741b709cd686fca87 (patch) | |
tree | e339593bb4e3858fa9379d14752dc52bf5949825 /text_recognizer/data | |
parent | ae8bfa62f0e02bd70c27bc1e71697249a5a79e7e (diff) |
Remove Barlow Twins
Diffstat (limited to 'text_recognizer/data')
-rw-r--r-- | text_recognizer/data/transforms/barlow.py | 19 |
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 |