diff --git a/bob/ip/binseg/configs/datasets/chasedb1/covd.py b/bob/ip/binseg/configs/datasets/chasedb1/covd.py index 970caf2fa82b75a9bf4bfd330e9071703b082b80..a2f9213172393e4def6dd7e4b23018b8d68c6e3e 100644 --- a/bob/ip/binseg/configs/datasets/chasedb1/covd.py +++ b/bob/ip/binseg/configs/datasets/chasedb1/covd.py @@ -27,13 +27,26 @@ from bob.ip.binseg.configs.datasets.chasedb1.first_annotator import ( dataset = dict(**_baseline) dataset["__train__"] = ConcatDataset( [ - _augment(_mtest["drive"], rotation_before=True), - _augment(_mtest["stare"], rotation_before=True), - _augment(_mtest["hrf"], rotation_before=False), - _augment(_mtest["iostar"], rotation_before=False), + _augment(_mtest["drive (train)"], rotation_before=True), + _augment(_mtest["drive (test)"], rotation_before=True), + _augment(_mtest["stare (train)"], rotation_before=True), + _augment(_mtest["stare (test)"], rotation_before=True), + _augment(_mtest["hrf (train)"], rotation_before=False), + _augment(_mtest["hrf (test)"], rotation_before=False), + _augment(_mtest["iostar (train)"], rotation_before=False), + _augment(_mtest["iostar (test)"], rotation_before=False), ] ) dataset["train"] = ConcatDataset( - [_mtest["drive"], _mtest["stare"], _mtest["hrf"], _mtest["iostar"],] + [ + _mtest["drive (train)"], + _mtest["drive (test)"], + _mtest["stare (train)"], + _mtest["stare (test)"], + _mtest["hrf (train)"], + _mtest["hrf (test)"], + _mtest["iostar (train)"], + _mtest["iostar (test)"], + ] ) dataset["__valid__"] = dataset["train"] diff --git a/bob/ip/binseg/configs/datasets/chasedb1/mtest.py b/bob/ip/binseg/configs/datasets/chasedb1/mtest.py index fdee4479ad4c111d5450dd4741a3b94b630942e0..20731847ffcf9b4914085697dec4fa240235e1f1 100644 --- a/bob/ip/binseg/configs/datasets/chasedb1/mtest.py +++ b/bob/ip/binseg/configs/datasets/chasedb1/mtest.py @@ -15,10 +15,22 @@ from bob.ip.binseg.configs.datasets.chasedb1.xtest import ( dataset = { "train": _xt["train"], "test": _xt["test"], - "drive": _xt["drive"].copy([CenterCrop((544, 544)), Resize(960)]), - "stare": _xt["stare"].copy( + "drive (train)": _xt["drive (train)"].copy( + [CenterCrop((544, 544)), Resize(960)] + ), + "drive (test)": _xt["drive (test)"].copy( + [CenterCrop((544, 544)), Resize(960)] + ), + "stare (train)": _xt["stare (train)"].copy( [Pad((0, 32, 0, 32)), Resize(960), CenterCrop(960)] ), - "hrf": _xt["hrf"].copy([Pad((0, 584, 0, 584)), Resize(960)]), - "iostar": _xt["iostar"].copy([Resize(960)]), + "stare (test)": _xt["stare (test)"].copy( + [Pad((0, 32, 0, 32)), Resize(960), CenterCrop(960)] + ), + "hrf (train)": _xt["hrf (train)"].copy( + [Pad((0, 584, 0, 584)), Resize(960)] + ), + "hrf (test)": _xt["hrf (test)"].copy([Pad((0, 584, 0, 584)), Resize(960)]), + "iostar (train)": _xt["iostar (train)"].copy([Resize(960)]), + "iostar (test)": _xt["iostar (test)"].copy([Resize(960)]), } diff --git a/bob/ip/binseg/configs/datasets/chasedb1/xtest.py b/bob/ip/binseg/configs/datasets/chasedb1/xtest.py index baf44c1b11d34ea868680235cc8f52c893b6f756..586b142c97cedb7684fc9e55fef2ec0b7b150026 100644 --- a/bob/ip/binseg/configs/datasets/chasedb1/xtest.py +++ b/bob/ip/binseg/configs/datasets/chasedb1/xtest.py @@ -16,8 +16,12 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar dataset = { "train": _chase["train"], "test": _chase["test"], - "drive": _drive["test"], - "stare": _stare["test"], - "hrf": _hrf["test"], - "iostar": _iostar["test"], + "drive (train)": _drive["train"], + "drive (test)": _drive["test"], + "stare (train)": _stare["train"], + "stare (test)": _stare["test"], + "hrf (train)": _hrf["train"], + "hrf (test)": _hrf["test"], + "iostar (train)": _iostar["train"], + "iostar (test)": _iostar["test"], } diff --git a/bob/ip/binseg/configs/datasets/drive/covd.py b/bob/ip/binseg/configs/datasets/drive/covd.py index 5010b35dbc50c49a53891e5d012f514f49385022..c32be4f31070a5f10020750dd30b2dca71e5b9b2 100644 --- a/bob/ip/binseg/configs/datasets/drive/covd.py +++ b/bob/ip/binseg/configs/datasets/drive/covd.py @@ -27,13 +27,26 @@ from bob.ip.binseg.configs.datasets.drive.default import ( dataset = dict(**_baseline) dataset["__train__"] = ConcatDataset( [ - _augment(_mtest["stare"], rotation_before=True), - _augment(_mtest["chasedb1"], rotation_before=False), - _augment(_mtest["hrf"], rotation_before=False), - _augment(_mtest["iostar"], rotation_before=False), + _augment(_mtest["stare (train)"], rotation_before=True), + _augment(_mtest["stare (test)"], rotation_before=True), + _augment(_mtest["chasedb1 (train)"], rotation_before=False), + _augment(_mtest["chasedb1 (test)"], rotation_before=False), + _augment(_mtest["hrf (train)"], rotation_before=False), + _augment(_mtest["hrf (test)"], rotation_before=False), + _augment(_mtest["iostar (train)"], rotation_before=False), + _augment(_mtest["iostar (test)"], rotation_before=False), ] ) dataset["train"] = ConcatDataset( - [_mtest["stare"], _mtest["chasedb1"], _mtest["hrf"], _mtest["iostar"],] + [ + _mtest["stare (train)"], + _mtest["stare (test)"], + _mtest["chasedb1 (train)"], + _mtest["chasedb1 (test)"], + _mtest["hrf (train)"], + _mtest["hrf (test)"], + _mtest["iostar (train)"], + _mtest["iostar (test)"], + ] ) dataset["__valid__"] = dataset["train"] diff --git a/bob/ip/binseg/configs/datasets/drive/mtest.py b/bob/ip/binseg/configs/datasets/drive/mtest.py index c6ef0eb5d186cfce8b337ec8c41cea89c5e76245..624cfff800177ab814a83d3190d0725fa6063f3a 100644 --- a/bob/ip/binseg/configs/datasets/drive/mtest.py +++ b/bob/ip/binseg/configs/datasets/drive/mtest.py @@ -15,8 +15,20 @@ from bob.ip.binseg.configs.datasets.drive.xtest import ( dataset = { "train": _xt["train"], "test": _xt["test"], - "stare": _xt["stare"].copy([Resize(471), Pad((0, 37, 0, 36))]), - "chasedb1": _xt["chasedb1"].copy([Resize(544), Crop(0, 12, 544, 544)]), - "hrf": _xt["hrf"].copy([Resize((363)), Pad((0, 90, 0, 91))]), - "iostar": _xt["iostar"].copy([Resize(544)]), + "stare (train)": _xt["stare (train)"].copy( + [Resize(471), Pad((0, 37, 0, 36))] + ), + "stare (test)": _xt["stare (test)"].copy( + [Resize(471), Pad((0, 37, 0, 36))] + ), + "chasedb1 (train)": _xt["chasedb1 (train)"].copy( + [Resize(544), Crop(0, 12, 544, 544)] + ), + "chasedb1 (test)": _xt["chasedb1 (test)"].copy( + [Resize(544), Crop(0, 12, 544, 544)] + ), + "hrf (train)": _xt["hrf (train)"].copy([Resize((363)), Pad((0, 90, 0, 91))]), + "hrf (test)": _xt["hrf (test)"].copy([Resize((363)), Pad((0, 90, 0, 91))]), + "iostar (train)": _xt["iostar (train)"].copy([Resize(544)]), + "iostar (test)": _xt["iostar (test)"].copy([Resize(544)]), } diff --git a/bob/ip/binseg/configs/datasets/drive/xtest.py b/bob/ip/binseg/configs/datasets/drive/xtest.py index b04e52328fe536493f3ae5b0772779cb3827413d..ced90727aa2e585ee0571b691d2681ee8188521b 100644 --- a/bob/ip/binseg/configs/datasets/drive/xtest.py +++ b/bob/ip/binseg/configs/datasets/drive/xtest.py @@ -18,8 +18,12 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar dataset = { "train": _drive["train"], "test": _drive["test"], - "stare": _stare["test"], - "chasedb1": _chase["test"], - "hrf": _hrf["test"], - "iostar": _iostar["test"], + "stare (train)": _stare["train"], + "stare (test)": _stare["test"], + "chasedb1 (train)": _chase["train"], + "chasedb1 (test)": _chase["test"], + "hrf (train)": _hrf["train"], + "hrf (test)": _hrf["test"], + "iostar (train)": _iostar["train"], + "iostar (test)": _iostar["test"], } diff --git a/bob/ip/binseg/configs/datasets/hrf/covd.py b/bob/ip/binseg/configs/datasets/hrf/covd.py index 4eeb70936755f2d57860e6ef89910fb0cb09eaa2..df7be2582186344861c78c7980b0e4b57c1cf573 100644 --- a/bob/ip/binseg/configs/datasets/hrf/covd.py +++ b/bob/ip/binseg/configs/datasets/hrf/covd.py @@ -24,13 +24,26 @@ from bob.ip.binseg.configs.datasets.hrf.default import dataset as _baseline dataset = dict(**_baseline) dataset["__train__"] = ConcatDataset( [ - _augment(_mtest["drive"], rotation_before=True), - _augment(_mtest["stare"], rotation_before=True), - _augment(_mtest["chasedb1"], rotation_before=True), - _augment(_mtest["iostar"], rotation_before=True), + _augment(_mtest["drive (train)"], rotation_before=True), + _augment(_mtest["drive (test)"], rotation_before=True), + _augment(_mtest["stare (train)"], rotation_before=True), + _augment(_mtest["stare (test)"], rotation_before=True), + _augment(_mtest["chasedb1 (train)"], rotation_before=True), + _augment(_mtest["chasedb1 (test)"], rotation_before=True), + _augment(_mtest["iostar (train)"], rotation_before=True), + _augment(_mtest["iostar (test)"], rotation_before=True), ] ) dataset["train"] = ConcatDataset( - [_mtest["drive"], _mtest["stare"], _mtest["chasedb1"], _mtest["iostar"],] + [ + _mtest["drive (train)"], + _mtest["drive (test)"], + _mtest["stare (train)"], + _mtest["stare (test)"], + _mtest["chasedb1 (train)"], + _mtest["chasedb1 (test)"], + _mtest["iostar (train)"], + _mtest["iostar (test)"], + ] ) dataset["__valid__"] = dataset["train"] diff --git a/bob/ip/binseg/configs/datasets/hrf/mtest.py b/bob/ip/binseg/configs/datasets/hrf/mtest.py index b5f8f646f4000ddcd89405e379899f45ec5c56b8..b73e64b00423a21bdf0ff8cb4fce2a383023c2c7 100644 --- a/bob/ip/binseg/configs/datasets/hrf/mtest.py +++ b/bob/ip/binseg/configs/datasets/hrf/mtest.py @@ -12,14 +12,28 @@ from bob.ip.binseg.configs.datasets.hrf.xtest import dataset as _xt dataset = { "train": _xt["train"], "test": _xt["test"], - "drive": _xt["drive"].copy( + "drive (train)": _xt["drive (train)"].copy( [Crop(75, 10, 416, 544), Pad((21, 0, 22, 0)), Resize(1168)] ), - "stare": _xt["stare"].copy( + "drive (test)": _xt["drive (test)"].copy( + [Crop(75, 10, 416, 544), Pad((21, 0, 22, 0)), Resize(1168)] + ), + "stare (train)": _xt["stare (train)"].copy( + [Crop(50, 0, 500, 705), Resize(1168), Pad((1, 0, 1, 0))] + ), + "stare (test)": _xt["stare (test)"].copy( [Crop(50, 0, 500, 705), Resize(1168), Pad((1, 0, 1, 0))] ), - "chasedb1": _xt["chasedb1"].copy([Crop(140, 18, 680, 960), Resize(1168)]), - "iostar": _xt["iostar"].copy( + "chasedb1 (train)": _xt["chasedb1 (train)"].copy( + [Crop(140, 18, 680, 960), Resize(1168)] + ), + "chasedb1 (test)": _xt["chasedb1 (test)"].copy( + [Crop(140, 18, 680, 960), Resize(1168)] + ), + "iostar (train)": _xt["iostar (train)"].copy( + [Crop(144, 0, 768, 1024), Pad((30, 0, 30, 0)), Resize(1168)] + ), + "iostar (test)": _xt["iostar (test)"].copy( [Crop(144, 0, 768, 1024), Pad((30, 0, 30, 0)), Resize(1168)] ), } diff --git a/bob/ip/binseg/configs/datasets/hrf/xtest.py b/bob/ip/binseg/configs/datasets/hrf/xtest.py index 6f96074fb9709f38d4ce89296e242e065520676f..277537a2bca89723ab97edb2c4663a179ce508fa 100644 --- a/bob/ip/binseg/configs/datasets/hrf/xtest.py +++ b/bob/ip/binseg/configs/datasets/hrf/xtest.py @@ -15,8 +15,12 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar dataset = { "train": _hrf["train"], "test": _hrf["test"], - "drive": _drive["test"], - "stare": _stare["test"], - "chasedb1": _chase["test"], - "iostar": _iostar["test"], + "drive (train)": _drive["train"], + "drive (test)": _drive["test"], + "stare (train)": _stare["train"], + "stare (test)": _stare["test"], + "chasedb1 (train)": _chase["train"], + "chasedb1 (test)": _chase["test"], + "iostar (train)": _iostar["train"], + "iostar (test)": _iostar["test"], } diff --git a/bob/ip/binseg/configs/datasets/iostar/covd.py b/bob/ip/binseg/configs/datasets/iostar/covd.py index 9b253e3a9be7a24101b8328540a96021e6c3cc05..cc1802a869e8286fdf5538ac9d8946648ce47f2f 100644 --- a/bob/ip/binseg/configs/datasets/iostar/covd.py +++ b/bob/ip/binseg/configs/datasets/iostar/covd.py @@ -24,13 +24,26 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _baseline dataset = dict(**_baseline) dataset["__train__"] = ConcatDataset( [ - _augment(_mtest["drive"], rotation_before=True), - _augment(_mtest["stare"], rotation_before=True), - _augment(_mtest["chasedb1"], rotation_before=True), - _augment(_mtest["hrf"], rotation_before=False), + _augment(_mtest["drive (train)"], rotation_before=True), + _augment(_mtest["drive (test)"], rotation_before=True), + _augment(_mtest["stare (train)"], rotation_before=True), + _augment(_mtest["stare (test)"], rotation_before=True), + _augment(_mtest["chasedb1 (train)"], rotation_before=True), + _augment(_mtest["chasedb1 (test)"], rotation_before=True), + _augment(_mtest["hrf (train)"], rotation_before=False), + _augment(_mtest["hrf (test)"], rotation_before=False), ] ) dataset["train"] = ConcatDataset( - [_mtest["drive"], _mtest["stare"], _mtest["chasedb1"], _mtest["hrf"],] + [ + _mtest["drive (train)"], + _mtest["drive (test)"], + _mtest["stare (train)"], + _mtest["stare (test)"], + _mtest["chasedb1 (train)"], + _mtest["chasedb1 (test)"], + _mtest["hrf (train)"], + _mtest["hrf (test)"], + ] ) dataset["__valid__"] = dataset["train"] diff --git a/bob/ip/binseg/configs/datasets/iostar/vessel_mtest.py b/bob/ip/binseg/configs/datasets/iostar/vessel_mtest.py index c89cf204442060247639277b82a8e77affd05295..85c07e2ec56b4c5ecd0057f71efff314e1da2dda 100644 --- a/bob/ip/binseg/configs/datasets/iostar/vessel_mtest.py +++ b/bob/ip/binseg/configs/datasets/iostar/vessel_mtest.py @@ -12,10 +12,26 @@ from bob.ip.binseg.configs.datasets.iostar.vessel_xtest import dataset as _xt dataset = { "train": _xt["train"], "test": _xt["test"], - "drive": _xt["drive"].copy([CenterCrop((540, 540)), Resize(1024)]), - "stare": _xt["stare"].copy( + "drive (train)": _xt["drive (train)"].copy( + [CenterCrop((540, 540)), Resize(1024)] + ), + "drive (test)": _xt["drive (test)"].copy( + [CenterCrop((540, 540)), Resize(1024)] + ), + "stare (train)": _xt["stare (train)"].copy( + [Pad((0, 32, 0, 32)), Resize(1024), CenterCrop(1024)] + ), + "stare (test)": _xt["stare (test)"].copy( [Pad((0, 32, 0, 32)), Resize(1024), CenterCrop(1024)] ), - "chasedb1": _xt["chasedb1"].copy([Crop(0, 18, 960, 960), Resize(1024)]), - "hrf": _xt["hrf"].copy([Pad((0, 584, 0, 584)), Resize(1024)]), + "chasedb1 (train)": _xt["chasedb1 (train)"].copy( + [Crop(0, 18, 960, 960), Resize(1024)] + ), + "chasedb1 (test)": _xt["chasedb1 (test)"].copy( + [Crop(0, 18, 960, 960), Resize(1024)] + ), + "hrf (train)": _xt["hrf (train)"].copy( + [Pad((0, 584, 0, 584)), Resize(1024)] + ), + "hrf (test)": _xt["hrf (test)"].copy([Pad((0, 584, 0, 584)), Resize(1024)]), } diff --git a/bob/ip/binseg/configs/datasets/iostar/vessel_xtest.py b/bob/ip/binseg/configs/datasets/iostar/vessel_xtest.py index 0d6272751ee4dc916c6e46f0b55209e1263f4190..14541c799d45c19158ed202cd088f7d1121eec09 100644 --- a/bob/ip/binseg/configs/datasets/iostar/vessel_xtest.py +++ b/bob/ip/binseg/configs/datasets/iostar/vessel_xtest.py @@ -15,8 +15,12 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar dataset = { "train": _iostar["train"], "test": _iostar["test"], - "drive": _drive["test"], - "stare": _stare["test"], - "chasedb1": _chase["test"], - "hrf": _hrf["test"], + "drive (train)": _drive["train"], + "drive (test)": _drive["test"], + "stare (train)": _stare["train"], + "stare (test)": _stare["test"], + "chasedb1 (train)": _chase["train"], + "chasedb1 (test)": _chase["test"], + "hrf (train)": _hrf["train"], + "hrf (test)": _hrf["test"], } diff --git a/bob/ip/binseg/configs/datasets/stare/covd.py b/bob/ip/binseg/configs/datasets/stare/covd.py index ad1c2aa12da6ccc9e9c6a5e26f54818c77f6ccbf..57000eae6fd33bd83c2c2c61297a43fe39db6030 100644 --- a/bob/ip/binseg/configs/datasets/stare/covd.py +++ b/bob/ip/binseg/configs/datasets/stare/covd.py @@ -27,13 +27,26 @@ from bob.ip.binseg.configs.datasets.stare.ah import ( dataset = dict(**_baseline) dataset["__train__"] = ConcatDataset( [ - _augment(_mtest["drive"], rotation_before=True), - _augment(_mtest["chasedb1"], rotation_before=True), - _augment(_mtest["hrf"], rotation_before=False), - _augment(_mtest["iostar"], rotation_before=False), + _augment(_mtest["drive (train)"], rotation_before=True), + _augment(_mtest["drive (test)"], rotation_before=True), + _augment(_mtest["chasedb1 (train)"], rotation_before=True), + _augment(_mtest["chasedb1 (test)"], rotation_before=True), + _augment(_mtest["hrf (train)"], rotation_before=False), + _augment(_mtest["hrf (test)"], rotation_before=False), + _augment(_mtest["iostar (train)"], rotation_before=False), + _augment(_mtest["iostar (test)"], rotation_before=False), ] ) dataset["train"] = ConcatDataset( - [_mtest["drive"], _mtest["chasedb1"], _mtest["hrf"], _mtest["iostar"],] + [ + _mtest["drive (train)"], + _mtest["drive (test)"], + _mtest["chasedb1 (train)"], + _mtest["chasedb1 (test)"], + _mtest["hrf (train)"], + _mtest["hrf (test)"], + _mtest["iostar (train)"], + _mtest["iostar (test)"], + ] ) dataset["__valid__"] = dataset["train"] diff --git a/bob/ip/binseg/configs/datasets/stare/mtest.py b/bob/ip/binseg/configs/datasets/stare/mtest.py index 4a914487061d18bf379d14a79787c38403948c71..c8910f510697532904f40654708fcd1db7dbfb7c 100644 --- a/bob/ip/binseg/configs/datasets/stare/mtest.py +++ b/bob/ip/binseg/configs/datasets/stare/mtest.py @@ -15,10 +15,26 @@ from bob.ip.binseg.configs.datasets.stare.xtest import ( dataset = { "train": _xt["train"], "test": _xt["test"], - "drive": _xt["drive"].copy( + "drive (train)": _xt["drive (train)"].copy( [CenterCrop((470, 544)), Pad((10, 9, 10, 8)), Resize(608)] ), - "chasedb1": _xt["chasedb1"].copy([CenterCrop((829, 960)), Resize(608)]), - "hrf": _xt["hrf"].copy([Pad((0, 345, 0, 345)), Resize(608)]), - "iostar": _xt["iostar"].copy([Pad((81, 0, 81, 0)), Resize(608)]), + "drive (test)": _xt["drive (test)"].copy( + [CenterCrop((470, 544)), Pad((10, 9, 10, 8)), Resize(608)] + ), + "chasedb1 (train)": _xt["chasedb1 (train)"].copy( + [CenterCrop((829, 960)), Resize(608)] + ), + "chasedb1 (test)": _xt["chasedb1 (test)"].copy( + [CenterCrop((829, 960)), Resize(608)] + ), + "hrf (train)": _xt["hrf (train)"].copy( + [Pad((0, 345, 0, 345)), Resize(608)] + ), + "hrf (test)": _xt["hrf (test)"].copy([Pad((0, 345, 0, 345)), Resize(608)]), + "iostar (train)": _xt["iostar (train)"].copy( + [Pad((81, 0, 81, 0)), Resize(608)] + ), + "iostar (test)": _xt["iostar (test)"].copy( + [Pad((81, 0, 81, 0)), Resize(608)] + ), } diff --git a/bob/ip/binseg/configs/datasets/stare/xtest.py b/bob/ip/binseg/configs/datasets/stare/xtest.py index a36495e26d739fe64a48b1e17dc80fed75be7a42..6aef1f6cc1e9f0db4064be33291f06cf3739e5ea 100644 --- a/bob/ip/binseg/configs/datasets/stare/xtest.py +++ b/bob/ip/binseg/configs/datasets/stare/xtest.py @@ -18,8 +18,12 @@ from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar dataset = { "train": _stare["train"], "test": _stare["test"], - "drive": _drive["test"], - "chasedb1": _chase["test"], - "hrf": _hrf["test"], - "iostar": _iostar["test"], + "drive (train)": _drive["train"], + "drive (test)": _drive["test"], + "chasedb1 (train)": _chase["train"], + "chasedb1 (test)": _chase["test"], + "hrf (train)": _hrf["train"], + "hrf (test)": _hrf["test"], + "iostar (train)": _iostar["train"], + "iostar (test)": _iostar["test"], }