summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/convnext/downsample.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks/convnext/downsample.py')
-rw-r--r--text_recognizer/networks/convnext/downsample.py21
1 files changed, 0 insertions, 21 deletions
diff --git a/text_recognizer/networks/convnext/downsample.py b/text_recognizer/networks/convnext/downsample.py
deleted file mode 100644
index a8a0466..0000000
--- a/text_recognizer/networks/convnext/downsample.py
+++ /dev/null
@@ -1,21 +0,0 @@
-"""Convnext downsample module."""
-from typing import Tuple
-
-from einops.layers.torch import Rearrange
-from torch import Tensor, nn
-
-
-class Downsample(nn.Module):
- """Downsamples feature maps by patches."""
-
- def __init__(self, dim: int, dim_out: int, factors: Tuple[int, int]) -> None:
- super().__init__()
- s1, s2 = factors
- self.fn = nn.Sequential(
- Rearrange("b c (h s1) (w s2) -> b (c s1 s2) h w", s1=s1, s2=s2),
- nn.Conv2d(dim * s1 * s2, dim_out, 1),
- )
-
- def forward(self, x: Tensor) -> Tensor:
- """Applies patch function."""
- return self.fn(x)