PRIYANSHUDHAKED commited on
Commit
2da8d8c
·
verified ·
1 Parent(s): 639713f

Delete src/miocodec/pupu/modules/anti_aliasing/resample.py with huggingface_hub

Browse files
src/miocodec/pupu/modules/anti_aliasing/resample.py DELETED
@@ -1,55 +0,0 @@
1
- import torch.nn as nn
2
- from torch.nn import functional as F
3
-
4
- from .filter import *
5
-
6
-
7
- class UpSample1d(nn.Module):
8
- def __init__(self, ratio=2, kernel_size=None):
9
- super().__init__()
10
- self.ratio = ratio
11
- self.kernel_size = (
12
- int(6 * ratio // 2) * 2 if kernel_size is None else kernel_size
13
- )
14
- self.stride = ratio
15
- self.pad = self.kernel_size // ratio - 1
16
- self.pad_left = self.pad * self.stride + (self.kernel_size - self.stride) // 2
17
- self.pad_right = (
18
- self.pad * self.stride + (self.kernel_size - self.stride + 1) // 2
19
- )
20
- filter = kaiser_sinc_filter1d(
21
- cutoff=0.5 / ratio, half_width=0.6 / ratio, kernel_size=self.kernel_size
22
- )
23
- self.register_buffer("filter", filter)
24
-
25
- # x: [B, C, T]
26
- def forward(self, x):
27
- _, C, _ = x.shape
28
-
29
- x = F.pad(x, (self.pad, self.pad), mode="replicate")
30
- x = self.ratio * F.conv_transpose1d(
31
- x, self.filter.expand(C, -1, -1), stride=self.stride, groups=C
32
- )
33
- x = x[..., self.pad_left : -self.pad_right]
34
-
35
- return x
36
-
37
-
38
- class DownSample1d(nn.Module):
39
- def __init__(self, ratio=2, kernel_size=None):
40
- super().__init__()
41
- self.ratio = ratio
42
- self.kernel_size = (
43
- int(6 * ratio // 2) * 2 if kernel_size is None else kernel_size
44
- )
45
- self.lowpass = LowPassFilter1d(
46
- cutoff=0.5 / ratio,
47
- half_width=0.6 / ratio,
48
- stride=ratio,
49
- kernel_size=self.kernel_size,
50
- )
51
-
52
- def forward(self, x):
53
- xx = self.lowpass(x)
54
-
55
- return xx