diff --git a/src/mednet/config/data/hivtb/datamodule.py b/src/mednet/config/data/hivtb/datamodule.py index 7a5617330e86b9c8bdd2f57a0a20ed3902cafd90..68a7b7a3bdfea302ecd41bc750355b354b85c5f2 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 @@ -58,7 +58,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 e19f32009e7a3ea7935a1fa3aea325c493728310..86e9fdb7058c3e25351212b93be7a8167c9ecc03 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 @@ -57,7 +57,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 94743f68edec801d8e73c5badb69f28f45ae8beb..778d505f96813f28b348b9a251d91f4213871d77 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 @@ -59,7 +59,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 4b04f871849abc6e3c2e997300c4349f42cb0725..6853ebe56718c38db9e1e1d72a2c08ded11627dd 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 @@ -68,7 +68,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 e9d55cef980f681c078c29603a2f50906a2e316c..67846f6c7c0842c18f24d83a4018beb1ccc908b9 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 @@ -53,7 +53,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