[data.segment.cxr8] Fixes mask loading

src/mednet/data/segment/cxr8.py

@@ -53,7 +53,6 @@ import os
 import pathlib
 import typing
 
-import numpy as np
 import PIL.Image
 import torch
 from torchvision import tv_tensors
@@ -159,10 +158,9 @@ class RawDataLoader(BaseDataLoader):
                 )
             )
 
-        target = np.array(PIL.Image.open(self.datadir / target_path))
-        target = np.where(target == 255, 0, target)
-        target = PIL.Image.fromarray(np.array(target > 0))
-
+        target = PIL.Image.open(self.datadir / target_path).convert(
+            mode="1", dither=None
+        )
         return to_dtype(to_image(target), torch.float32, scale=True)