|
@@ -1604,7 +1604,6 @@ class HyperACE(nn.Module):
|
|
|
|
|
|
def forward(self, X):
|
|
def forward(self, X):
|
|
x = self.fuse(X)
|
|
x = self.fuse(X)
|
|
- """Forward pass through C2f layer."""
|
|
|
|
y = list(self.cv1(x).chunk(3, 1))
|
|
y = list(self.cv1(x).chunk(3, 1))
|
|
out1 = self.branch1(y[1])
|
|
out1 = self.branch1(y[1])
|
|
out2 = self.branch2(y[1])
|
|
out2 = self.branch2(y[1])
|
|
@@ -1631,4 +1630,4 @@ class FullPAD_Tunnel(nn.Module):
|
|
self.gate = nn.Parameter(torch.tensor(0.0))
|
|
self.gate = nn.Parameter(torch.tensor(0.0))
|
|
def forward(self, x):
|
|
def forward(self, x):
|
|
out = x[0] + self.gate * x[1]
|
|
out = x[0] + self.gate * x[1]
|
|
- return out
|
|
|
|
|
|
+ return out
|