From 7239bce214607c70a7a91358586f265b2f74de7b Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 3 Sep 2023 01:10:11 +0200 Subject: Delete convnext --- text_recognizer/network/convnext/downsample.py | 21 --------------------- 1 file changed, 21 deletions(-) delete mode 100644 text_recognizer/network/convnext/downsample.py (limited to 'text_recognizer/network/convnext/downsample.py') diff --git a/text_recognizer/network/convnext/downsample.py b/text_recognizer/network/convnext/downsample.py deleted file mode 100644 index a8a0466..0000000 --- a/text_recognizer/network/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) -- cgit v1.2.3-70-g09d2