diff --git a/src/mednet/config/data/hivtb/datamodule.py b/src/mednet/config/data/hivtb/datamodule.py index d5bf81038eb704705fc32e599edd721d6b3d18ca..206a3ec6630381452393831de109859e68eff556 100644 --- a/src/mednet/config/data/hivtb/datamodule.py +++ b/src/mednet/config/data/hivtb/datamodule.py @@ -12,7 +12,7 @@ import os import PIL.Image -from torchvision.transforms.functional import center_crop, to_tensor +from torchvision.transforms.functional import to_tensor from ....data.datamodule import CachingDataModule from ....data.image_utils import remove_black_borders @@ -59,7 +59,6 @@ class RawDataLoader(_BaseRawDataLoader): ) image = remove_black_borders(image) tensor = to_tensor(image) - tensor = center_crop(tensor, min(*tensor.shape[1:])) # use the code below to view generated images # from torchvision.transforms.functional import to_pil_image diff --git a/src/mednet/config/data/montgomery/datamodule.py b/src/mednet/config/data/montgomery/datamodule.py index ec1b7c1422a0ce4bd5948c7446c14f6d50478fb2..ff8d21d75948afff3263d18a2e5673b9cf778c11 100644 --- a/src/mednet/config/data/montgomery/datamodule.py +++ b/src/mednet/config/data/montgomery/datamodule.py @@ -11,7 +11,7 @@ import os import PIL.Image -from torchvision.transforms.functional import center_crop, to_tensor +from torchvision.transforms.functional import to_tensor from ....data.datamodule import CachingDataModule from ....data.image_utils import remove_black_borders @@ -58,7 +58,6 @@ class RawDataLoader(_BaseRawDataLoader): image = PIL.Image.open(os.path.join(self.datadir, sample[0])) image = remove_black_borders(image) tensor = to_tensor(image) - tensor = center_crop(tensor, min(*tensor.shape[1:])) # use the code below to view generated images # from torchvision.transforms.functional import to_pil_image diff --git a/src/mednet/config/data/padchest/datamodule.py b/src/mednet/config/data/padchest/datamodule.py index a065deceaf9676219f9afd7234a40f9d9124de0d..436193e86e9f58fddebcd2fac22ce0951e8968c4 100644 --- a/src/mednet/config/data/padchest/datamodule.py +++ b/src/mednet/config/data/padchest/datamodule.py @@ -12,7 +12,7 @@ import os import numpy import PIL.Image -from torchvision.transforms.functional import center_crop, to_tensor +from torchvision.transforms.functional import to_tensor from ....data.datamodule import CachingDataModule from ....data.image_utils import remove_black_borders @@ -60,7 +60,6 @@ class RawDataLoader(_BaseRawDataLoader): image = remove_black_borders(image) array = numpy.array(image).astype(numpy.float32) / 65535 tensor = to_tensor(array) - tensor = center_crop(tensor, min(*tensor.shape[1:])) # use the code below to view generated images # from torchvision.transforms.functional import to_pil_image diff --git a/src/mednet/config/data/shenzhen/datamodule.py b/src/mednet/config/data/shenzhen/datamodule.py index 6409179465984e94e52c29f12bfb7bcc50b09596..9fe82d47a3c024680207f8d16c76db19c541eebf 100644 --- a/src/mednet/config/data/shenzhen/datamodule.py +++ b/src/mednet/config/data/shenzhen/datamodule.py @@ -11,7 +11,7 @@ import os import PIL.Image -from torchvision.transforms.functional import center_crop, to_tensor +from torchvision.transforms.functional import to_tensor from ....data.datamodule import CachingDataModule from ....data.image_utils import remove_black_borders @@ -62,7 +62,6 @@ class RawDataLoader(_BaseRawDataLoader): ) image = remove_black_borders(image) tensor = to_tensor(image) - tensor = center_crop(tensor, min(*tensor.shape[1:])) # use the code below to view generated images # from torchvision.transforms.functional import to_pil_image diff --git a/src/mednet/config/data/tbpoc/datamodule.py b/src/mednet/config/data/tbpoc/datamodule.py index ffe59568fd9524962e534deac1a72fcc1408cedd..c0339747471728b482a6e4c4d9f808b739a809aa 100644 --- a/src/mednet/config/data/tbpoc/datamodule.py +++ b/src/mednet/config/data/tbpoc/datamodule.py @@ -7,7 +7,7 @@ import os import PIL.Image -from torchvision.transforms.functional import center_crop, to_tensor +from torchvision.transforms.functional import to_tensor from ....data.datamodule import CachingDataModule from ....data.image_utils import remove_black_borders @@ -54,7 +54,6 @@ class RawDataLoader(_BaseRawDataLoader): image = PIL.Image.open(os.path.join(self.datadir, sample[0])) image = remove_black_borders(image) tensor = to_tensor(image) - tensor = center_crop(tensor, min(*tensor.shape[1:])) # use the code below to view generated images # from torchvision.transforms.functional import to_pil_image