From 8504a5f8a09d124e463f1d1226831be95782b194 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 17 Sep 2024 18:36:39 +0200
Subject: [PATCH] [data.segment.shenzhen] Do not need to crop image to mask
---
src/mednet/data/segment/shenzhen.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/mednet/data/segment/shenzhen.py b/src/mednet/data/segment/shenzhen.py
index 02570d7b..96f073a2 100644
--- a/src/mednet/data/segment/shenzhen.py
+++ b/src/mednet/data/segment/shenzhen.py
@@ -67,7 +67,6 @@ import torch
from torchvision import tv_tensors
from torchvision.transforms.v2.functional import to_dtype, to_image
-from ...models.transforms import crop_image_to_mask
from ...utils.rc import load_rc
from ..datamodule import CachingDataModule
from ..split import JSONDatabaseSplit
@@ -116,8 +115,8 @@ class RawDataLoader(BaseDataLoader):
# use image as a base since target() can be overriden by child class
mask = torch.ones((1, image.shape[-2], image.shape[-1]), dtype=torch.float32)
- image = tv_tensors.Image(crop_image_to_mask(image, mask))
- target = tv_tensors.Mask(crop_image_to_mask(target, mask))
+ image = tv_tensors.Image(image)
+ target = tv_tensors.Mask(target)
mask = tv_tensors.Mask(mask)
return dict(image=image, target=target, mask=mask, name=sample[0])
--
GitLab