SiLU_beta(c2)

models/common.py

@@ -12,7 +12,7 @@
 from PIL import Image
 from torch.cuda import amp
 
-from utils.activations import AconC, MetaAconC, FReLU
+from utils.activations import AconC, MetaAconC, FReLU, SiLU_beta
 from utils.datasets import letterbox
 from utils.general import non_max_suppression, make_divisible, scale_coords, increment_path, xyxy2xywh, save_one_box
 from utils.plots import color_list, plot_one_box
@@ -48,7 +48,7 @@ def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True):  # ch_in, ch_out, k
         # self.act = AconC() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
         # self.act = MetaAconC() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
         # self.act = SiLU_beta() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
-        self.act = MetaAconC(c2) if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
+        self.act = SiLU_beta(c2) if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
 
     def forward(self, x):
         return self.act(self.bn(self.conv(x)))