Update total/Model.py

total/Model.py

@@ -62,16 +62,3 @@ class Unet_TS(network.Network):
         nb_class, in_channels = state_dict["Model"]["Unet_TS"]["Head.Conv.weight"].shape[:2]
         self["Head"].add_module("Conv", torch.nn.Conv3d(in_channels = in_channels, out_channels = nb_class, kernel_size = 1, stride = 1, padding = 0))
         super().load(state_dict, init, ema)
-
-class Combine(Reduction):
-
-    def __init__(self):
-        pass
-
-    def __call__(self, tensors: list[torch.Tensor]) -> torch.Tensor:
-        fg_all = torch.cat([p[:, 1:, ...] for p in tensors], dim=1)
-        sum_fg = fg_all.sum(dim=1, keepdim=True)
-        bg = (1.0 - sum_fg).clamp(min=1e-6)
-
-        probs = torch.cat([bg, fg_all], dim=1)
-        return probs / probs.sum(dim=1, keepdim=True)