extract bugfixes: vgg-obstructed model path + unet-dfl output shape

Author: torzdf

plugins/extract/mask/unet_dfl.py

@@ -68,7 +68,7 @@ def process(self, batch: np.ndarray) -> np.ndarray:
         -------
         The predicted masks from the plugin
         """
-        return self.from_torch(batch.transpose(0, 3, 1, 2)).transpose(0, 2, 3, 1)
+        return self.from_torch(batch.transpose(0, 3, 1, 2))[:, 0]
 
 
 class ConvBlock(nn.Module):

plugins/extract/mask/vgg_obstructed.py

@@ -39,7 +39,7 @@ def load_model(self) -> VGGObstructedModel:
         -------
         The loaded VGGObstructed model
         """
-        weights = GetModel("Nirkin_500_softmax_v2.pth", 8).model_path
+        weights = GetModel("Nirkin_500_softmax_v2.pth", 5).model_path
         assert isinstance(weights, str)
         return T.cast(VGGObstructedModel, self.load_torch_model(VGGObstructedModel(),
                                                                 weights,