Skip to content
Snippets Groups Projects
Commit 8504a5f8 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[data.segment.shenzhen] Do not need to crop image to mask

parent a604975e
No related branches found
No related tags found
1 merge request!64Add object detection
...@@ -67,7 +67,6 @@ import torch ...@@ -67,7 +67,6 @@ import torch
from torchvision import tv_tensors from torchvision import tv_tensors
from torchvision.transforms.v2.functional import to_dtype, to_image 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 ...utils.rc import load_rc
from ..datamodule import CachingDataModule from ..datamodule import CachingDataModule
from ..split import JSONDatabaseSplit from ..split import JSONDatabaseSplit
...@@ -116,8 +115,8 @@ class RawDataLoader(BaseDataLoader): ...@@ -116,8 +115,8 @@ class RawDataLoader(BaseDataLoader):
# use image as a base since target() can be overriden by child class # 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) mask = torch.ones((1, image.shape[-2], image.shape[-1]), dtype=torch.float32)
image = tv_tensors.Image(crop_image_to_mask(image, mask)) image = tv_tensors.Image(image)
target = tv_tensors.Mask(crop_image_to_mask(target, mask)) target = tv_tensors.Mask(target)
mask = tv_tensors.Mask(mask) mask = tv_tensors.Mask(mask)
return dict(image=image, target=target, mask=mask, name=sample[0]) return dict(image=image, target=target, mask=mask, name=sample[0])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment