diff --git a/bob/ip/binseg/configs/datasets/drive1024test.py b/bob/ip/binseg/configs/datasets/drive1024test.py deleted file mode 100644 index 54d1cf5cb9f71b000dbdbb737d6f6565fbf71a43..0000000000000000000000000000000000000000 --- a/bob/ip/binseg/configs/datasets/drive1024test.py +++ /dev/null @@ -1,22 +0,0 @@ -#!/usr/bin/env python -# coding=utf-8 - -"""DRIVE (training set) for Vessel Segmentation - -The DRIVE database has been established to enable comparative studies on -segmentation of blood vessels in retinal images. - -* Reference: [DRIVE-2004]_ -* Original resolution (height x width): 584 x 565 -* This configuration resolution: 1024 x 1024 (center-crop) -* Test samples: 20 -* Split reference: [DRIVE-2004]_ -""" - -from bob.ip.binseg.data.transforms import * -_transforms = Compose([CenterCrop((540, 540)), Resize(1024), ToTensor()]) - -from bob.ip.binseg.data.utils import DelayedSample2TorchDataset -from bob.ip.binseg.data.drive import dataset as drive -dataset = DelayedSample2TorchDataset(drive.subsets("default")["test"], - transform=_transforms) diff --git a/bob/ip/binseg/configs/datasets/iostarvessel544test.py b/bob/ip/binseg/configs/datasets/iostarvessel544test.py deleted file mode 100644 index 321579bfaa23da7efcd4e722bc92859b1c93ae99..0000000000000000000000000000000000000000 --- a/bob/ip/binseg/configs/datasets/iostarvessel544test.py +++ /dev/null @@ -1,16 +0,0 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- - -from bob.db.iostar import Database as IOSTAR -from bob.ip.binseg.data.transforms import * -from bob.ip.binseg.data.binsegdataset import BinSegDataset - -#### Config #### - -transforms = Compose([Resize(544), ToTensor()]) - -# bob.db.dataset init -bobdb = IOSTAR(protocol="default_vessel") - -# PyTorch dataset -dataset = BinSegDataset(bobdb, split="test", transform=transforms) diff --git a/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544.py b/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544.py index 91720a235c101bf854ab01bb61062222f0c1506b..701d6b08586959d39d97c8790950b1befb6adb74 100644 --- a/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544.py +++ b/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544.py @@ -1,7 +1,7 @@ #!/usr/bin/env python # coding=utf-8 -"""COVD-STARE (training set) for Vessel Segmentation +"""COVD-DRIVE (training set) for Vessel Segmentation """ from bob.ip.binseg.configs.datasets.stare544 import dataset as _stare diff --git a/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544ssldrive.py b/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544ssldrive.py index 2605dc8183660171ae4d14555f0e129f3a9e8658..c4ce12ca8f488bf37903b4dccd15ba7e51dd4333 100644 --- a/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544ssldrive.py +++ b/bob/ip/binseg/configs/datasets/starechasedb1iostarhrf544ssldrive.py @@ -9,7 +9,7 @@ The dataset available in this file is composed of STARE, CHASE-DB1, IOSTAR vessel and HRF (with annotated samples) and DRIVE without labels. """ -from bob.ip.binseg.configs.datasets.starechasedb1iostarhrf544 import dataset as _unlabelled +from bob.ip.binseg.configs.datasets.starechasedb1iostarhrf544 import dataset as _labelled from bob.ip.binseg.configs.datasets.drive import dataset as _unlabelled from bob.ip.binseg.data.utils import SSLDataset dataset = SSLDataset(_labelled, _unlabelled)