diff --git a/src/ptbench/configs/datasets/tbx11k_simplified/__init__.py b/src/ptbench/configs/datasets/tbx11k_simplified/__init__.py index f52f2db110f7ca533d32e300f2309ffab3f14584..595c323178507ec04f9dc1768e080310b9ca7d50 100644 --- a/src/ptbench/configs/datasets/tbx11k_simplified/__init__.py +++ b/src/ptbench/configs/datasets/tbx11k_simplified/__init__.py @@ -7,7 +7,7 @@ def _maker(protocol, RGB=False): from torchvision import transforms from ....data.tbx11k_simplified import dataset as raw - from ....data.transforms import ElasticDeformation, RGBtoGreyscale8bit + from ....data.transforms import ElasticDeformation from .. import make_dataset as mk post_transforms = []