From 7001fb36dfd6d91db52446026ba54fd72d827d6f Mon Sep 17 00:00:00 2001
From: Driss KHALIL <dkhalil@vws072.idiap.ch>
Date: Fri, 19 Nov 2021 13:21:01 +0100
Subject: [PATCH 1/2] Add new protocols to datasets

---
 .../configs/datasets/chasedb1/__init__.py      |  9 +++++++++
 .../datasets/chasedb1/first_annotator_768.py   | 12 ++++++++++++
 .../configs/datasets/drionsdb/__init__.py      |  9 +++++++++
 .../configs/datasets/drionsdb/expert1_512.py   | 12 ++++++++++++
 .../configs/datasets/drionsdb/expert2_512.py   | 12 ++++++++++++
 .../configs/datasets/drishtigs1/__init__.py    | 13 +++++++++++++
 .../configs/datasets/drishtigs1/cup_all_512.py | 12 ++++++++++++
 .../datasets/drishtigs1/disc_all_512.py        | 12 ++++++++++++
 .../binseg/configs/datasets/drive/__init__.py  |  9 +++++++++
 .../configs/datasets/drive/default_768.py      | 12 ++++++++++++
 bob/ip/binseg/configs/datasets/hrf/__init__.py |  9 +++++++++
 .../binseg/configs/datasets/hrf/default_768.py | 12 ++++++++++++
 .../binseg/configs/datasets/iostar/__init__.py | 18 ++++++++++++++++++
 .../configs/datasets/iostar/optic_disc_512.py  | 12 ++++++++++++
 .../configs/datasets/iostar/vessel_768.py      | 12 ++++++++++++
 .../binseg/configs/datasets/refuge/__init__.py | 15 +++++++++++++++
 .../binseg/configs/datasets/refuge/cup_512.py  | 12 ++++++++++++
 .../binseg/configs/datasets/refuge/disc_512.py | 12 ++++++++++++
 .../configs/datasets/rimoner3/__init__.py      |  9 +++++++++
 .../configs/datasets/rimoner3/cup_exp1_512.py  | 12 ++++++++++++
 .../configs/datasets/rimoner3/disc_exp1_512.py | 12 ++++++++++++
 .../binseg/configs/datasets/stare/__init__.py  | 11 +++++++++++
 bob/ip/binseg/configs/datasets/stare/ah_768.py | 11 +++++++++++
 setup.py                                       | 14 ++++++++++++++
 24 files changed, 283 insertions(+)
 create mode 100644 bob/ip/binseg/configs/datasets/chasedb1/first_annotator_768.py
 create mode 100644 bob/ip/binseg/configs/datasets/drionsdb/expert1_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/drionsdb/expert2_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/drishtigs1/cup_all_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/drishtigs1/disc_all_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/drive/default_768.py
 create mode 100644 bob/ip/binseg/configs/datasets/hrf/default_768.py
 create mode 100644 bob/ip/binseg/configs/datasets/iostar/optic_disc_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/iostar/vessel_768.py
 create mode 100644 bob/ip/binseg/configs/datasets/refuge/cup_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/refuge/disc_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/rimoner3/cup_exp1_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/rimoner3/disc_exp1_512.py
 create mode 100644 bob/ip/binseg/configs/datasets/stare/ah_768.py

diff --git a/bob/ip/binseg/configs/datasets/chasedb1/__init__.py b/bob/ip/binseg/configs/datasets/chasedb1/__init__.py
index 571f314..287fda3 100644
--- a/bob/ip/binseg/configs/datasets/chasedb1/__init__.py
+++ b/bob/ip/binseg/configs/datasets/chasedb1/__init__.py
@@ -9,3 +9,12 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [Crop(0, 18, 960, 960)])
+
+
+def _maker_square(protocol):
+
+    from ....data.chasedb1 import dataset as raw
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((1, 20, 0, 20)), Resize((768, 768))])
diff --git a/bob/ip/binseg/configs/datasets/chasedb1/first_annotator_768.py b/bob/ip/binseg/configs/datasets/chasedb1/first_annotator_768.py
new file mode 100644
index 0000000..7afe056
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/chasedb1/first_annotator_768.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""CHASE-DB1 dataset for Vessel Segmentation
+
+Configuration resolution: 768 x 768 (after Pad and resize)
+
+"""
+
+from bob.ip.binseg.configs.datasets.chasedb1 import _maker_square
+
+dataset = _maker_square("first-annotator")
diff --git a/bob/ip/binseg/configs/datasets/drionsdb/__init__.py b/bob/ip/binseg/configs/datasets/drionsdb/__init__.py
index bd98b41..29a7977 100644
--- a/bob/ip/binseg/configs/datasets/drionsdb/__init__.py
+++ b/bob/ip/binseg/configs/datasets/drionsdb/__init__.py
@@ -9,3 +9,12 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [Pad((4, 8, 4, 8))])
+
+
+def _maker_square(protocol):
+
+    from ....data.drionsdb import dataset as raw
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((0, 100)), Resize((512, 512))])
diff --git a/bob/ip/binseg/configs/datasets/drionsdb/expert1_512.py b/bob/ip/binseg/configs/datasets/drionsdb/expert1_512.py
new file mode 100644
index 0000000..8d97c6a
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/drionsdb/expert1_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRIONS-DB for Optic Disc Segmentation (expert #1 annotations)
+
+Configuration resolution: 512x512 (after padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drionsdb import _maker_square
+
+dataset = _maker_square("expert1")
diff --git a/bob/ip/binseg/configs/datasets/drionsdb/expert2_512.py b/bob/ip/binseg/configs/datasets/drionsdb/expert2_512.py
new file mode 100644
index 0000000..df1dcc9
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/drionsdb/expert2_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRIONS-DB for Optic Disc Segmentation (expert #2 annotations)
+
+Configuration resolution: 512x512 (after padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drionsdb import _maker_square
+
+dataset = _maker_square("expert2")
diff --git a/bob/ip/binseg/configs/datasets/drishtigs1/__init__.py b/bob/ip/binseg/configs/datasets/drishtigs1/__init__.py
index dd2149b..c338cda 100644
--- a/bob/ip/binseg/configs/datasets/drishtigs1/__init__.py
+++ b/bob/ip/binseg/configs/datasets/drishtigs1/__init__.py
@@ -9,3 +9,16 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [ccrop((1760, 2048))])
+
+
+def _maker_square(protocol):
+
+    from ....data.drishtigs1 import dataset as raw
+    from ....data.transforms import CenterCrop as ccrop
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(
+        raw.subsets(protocol),
+        [ccrop((1760, 2048)), Pad((0, 144)), Resize((512, 512))],
+    )
diff --git a/bob/ip/binseg/configs/datasets/drishtigs1/cup_all_512.py b/bob/ip/binseg/configs/datasets/drishtigs1/cup_all_512.py
new file mode 100644
index 0000000..11d4e5c
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/drishtigs1/cup_all_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRISHTI-GS1 dataset for Cup Segmentation (agreed by all annotators)
+
+* Configuration resolution: 512 x 512 (after center cropping, padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drishtigs1 import _maker_square
+
+dataset = _maker_square("optic-cup-all")
diff --git a/bob/ip/binseg/configs/datasets/drishtigs1/disc_all_512.py b/bob/ip/binseg/configs/datasets/drishtigs1/disc_all_512.py
new file mode 100644
index 0000000..fce36f5
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/drishtigs1/disc_all_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRISHTI-GS1 dataset for Optic Disc Segmentation (agreed by all annotators)
+
+* Configuration resolution: 512 x 512 (after center cropping, padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drishtigs1 import _maker_square
+
+dataset = _maker_square("optic-disc-all")
diff --git a/bob/ip/binseg/configs/datasets/drive/__init__.py b/bob/ip/binseg/configs/datasets/drive/__init__.py
index 8037e9e..ca9e465 100644
--- a/bob/ip/binseg/configs/datasets/drive/__init__.py
+++ b/bob/ip/binseg/configs/datasets/drive/__init__.py
@@ -9,3 +9,12 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [ccrop((544, 544))])
+
+
+def _maker_square(protocol):
+
+    from ....data.drive import dataset as raw
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((10, 1, 10, 0)), Resize((768, 768))])
diff --git a/bob/ip/binseg/configs/datasets/drive/default_768.py b/bob/ip/binseg/configs/datasets/drive/default_768.py
new file mode 100644
index 0000000..9467bec
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/drive/default_768.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRIVE dataset for Vessel Segmentation (Resolution used for MTL models)
+
+This configuration resolution: 768 x 768 (Pad + resize)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drive import _maker_square
+
+dataset = _maker_square("default")
diff --git a/bob/ip/binseg/configs/datasets/hrf/__init__.py b/bob/ip/binseg/configs/datasets/hrf/__init__.py
index ef8285c..4d799c2 100644
--- a/bob/ip/binseg/configs/datasets/hrf/__init__.py
+++ b/bob/ip/binseg/configs/datasets/hrf/__init__.py
@@ -18,3 +18,12 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [Crop(0, 108, 2336, 3296)])
+
+
+def _maker_square(protocol):
+
+    from ....data.hrf import dataset as raw
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((0, 584)), Resize((768, 768))])
diff --git a/bob/ip/binseg/configs/datasets/hrf/default_768.py b/bob/ip/binseg/configs/datasets/hrf/default_768.py
new file mode 100644
index 0000000..28c0e9b
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/hrf/default_768.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""HRF dataset for Vessel Segmentation
+
+Configuration resolution: 768 x 768 (Pad + Resize)
+
+"""
+
+from bob.ip.binseg.configs.datasets.hrf import _maker_square
+
+dataset = _maker_square("default")
diff --git a/bob/ip/binseg/configs/datasets/iostar/__init__.py b/bob/ip/binseg/configs/datasets/iostar/__init__.py
index b3edfb7..f520443 100644
--- a/bob/ip/binseg/configs/datasets/iostar/__init__.py
+++ b/bob/ip/binseg/configs/datasets/iostar/__init__.py
@@ -8,3 +8,21 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [])
+
+
+def _maker_vessel_square(protocol):
+
+    from ....data.iostar import dataset as raw
+    from ....data.transforms import Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Resize((768, 768))])
+
+
+def _maker_od_square(protocol):
+
+    from ....data.iostar import dataset as raw
+    from ....data.transforms import Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Resize((512, 512))])
diff --git a/bob/ip/binseg/configs/datasets/iostar/optic_disc_512.py b/bob/ip/binseg/configs/datasets/iostar/optic_disc_512.py
new file mode 100644
index 0000000..ff84d19
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/iostar/optic_disc_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""IOSTAR dataset for Optic Disc Segmentation
+
+Configuration resolution: 512 x 512 (Resized )
+
+"""
+
+from bob.ip.binseg.configs.datasets.iostar import _maker_od_square
+
+dataset = _maker_od_square("optic-disc")
diff --git a/bob/ip/binseg/configs/datasets/iostar/vessel_768.py b/bob/ip/binseg/configs/datasets/iostar/vessel_768.py
new file mode 100644
index 0000000..9cb175b
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/iostar/vessel_768.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""IOSTAR dataset for Vessel Segmentation (default protocol)
+
+Configuration resolution: 768 x 768 (Resize)
+
+"""
+
+from bob.ip.binseg.configs.datasets.iostar import _maker_vessel_square
+
+dataset = _maker_vessel_square("vessel")
diff --git a/bob/ip/binseg/configs/datasets/refuge/__init__.py b/bob/ip/binseg/configs/datasets/refuge/__init__.py
index cfd1afb..7396585 100644
--- a/bob/ip/binseg/configs/datasets/refuge/__init__.py
+++ b/bob/ip/binseg/configs/datasets/refuge/__init__.py
@@ -15,3 +15,18 @@ def _maker(protocol):
     # and we keep the "train" set with the right transforms
     retval["train"] = train["train"]
     return retval
+
+
+def _maker_square(protocol):
+
+    from ....data.refuge import dataset as raw
+    from ....data.transforms import CenterCrop, Pad, Resize
+    from .. import make_dataset as mk
+
+    # due to different sizes, we need to make the dataset twice
+    train = mk(raw.subsets(protocol), [Resize(1539), Pad((21, 46, 22, 47))])
+    # we'll keep "dev" and "test" from the next one
+    retval = mk(raw.subsets(protocol), [CenterCrop(1632), Resize((512, 512))])
+    # and we keep the "train" set with the right transforms
+    retval["train"] = train["train"]
+    return retval
diff --git a/bob/ip/binseg/configs/datasets/refuge/cup_512.py b/bob/ip/binseg/configs/datasets/refuge/cup_512.py
new file mode 100644
index 0000000..e3a8a14
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/refuge/cup_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""REFUGE dataset for Optic Cup Segmentation
+
+* Configuration resolution: 512 x 512 (after resizing and padding)
+
+"""
+
+from bob.ip.binseg.configs.datasets.refuge import _maker_square
+
+dataset = _maker_square("optic-cup")
diff --git a/bob/ip/binseg/configs/datasets/refuge/disc_512.py b/bob/ip/binseg/configs/datasets/refuge/disc_512.py
new file mode 100644
index 0000000..fce36f5
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/refuge/disc_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""DRISHTI-GS1 dataset for Optic Disc Segmentation (agreed by all annotators)
+
+* Configuration resolution: 512 x 512 (after center cropping, padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.drishtigs1 import _maker_square
+
+dataset = _maker_square("optic-disc-all")
diff --git a/bob/ip/binseg/configs/datasets/rimoner3/__init__.py b/bob/ip/binseg/configs/datasets/rimoner3/__init__.py
index e813daf..3aefd9b 100644
--- a/bob/ip/binseg/configs/datasets/rimoner3/__init__.py
+++ b/bob/ip/binseg/configs/datasets/rimoner3/__init__.py
@@ -9,3 +9,12 @@ def _maker(protocol):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [Pad((8, 8, 8, 8))])
+
+
+def _maker_square(protocol):
+
+    from ....data.rimoner3 import dataset as raw
+    from ....data.transforms import Pad, Resize
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((176, 0)), Resize((512, 512))])
diff --git a/bob/ip/binseg/configs/datasets/rimoner3/cup_exp1_512.py b/bob/ip/binseg/configs/datasets/rimoner3/cup_exp1_512.py
new file mode 100644
index 0000000..e2a4132
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/rimoner3/cup_exp1_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""RIM-ONE r3 for Optic Cup Segmentation (expert #1 annotations)
+
+Configuration resolution: 512 x 512 (after padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.rimoner3 import _maker_square
+
+dataset = _maker_square("optic-cup-exp1")
diff --git a/bob/ip/binseg/configs/datasets/rimoner3/disc_exp1_512.py b/bob/ip/binseg/configs/datasets/rimoner3/disc_exp1_512.py
new file mode 100644
index 0000000..1e0b37c
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/rimoner3/disc_exp1_512.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""RIM-ONE r3 for Optic Disc Segmentation (expert #1 annotations)
+
+Configuration resolution: 512 x 512 (after padding and resizing)
+
+"""
+
+from bob.ip.binseg.configs.datasets.rimoner3 import _maker_square
+
+dataset = _maker_square("optic-disc-exp1")
diff --git a/bob/ip/binseg/configs/datasets/stare/__init__.py b/bob/ip/binseg/configs/datasets/stare/__init__.py
index e1929b0..7b43cfc 100644
--- a/bob/ip/binseg/configs/datasets/stare/__init__.py
+++ b/bob/ip/binseg/configs/datasets/stare/__init__.py
@@ -11,3 +11,14 @@ def _maker(protocol, raw=None):
     from .. import make_dataset as mk
 
     return mk(raw.subsets(protocol), [Pad((2, 1, 2, 2))])
+
+
+def _maker_square(protocol, raw=None):
+
+    from ....data.stare import dataset as _raw
+    from ....data.transforms import Pad, Resize
+
+    raw = raw or _raw  # allows user to recreate dataset for testing purposes
+    from .. import make_dataset as mk
+
+    return mk(raw.subsets(protocol), [Pad((1, 48, 0, 48)), Resize((768, 768))])
diff --git a/bob/ip/binseg/configs/datasets/stare/ah_768.py b/bob/ip/binseg/configs/datasets/stare/ah_768.py
new file mode 100644
index 0000000..59e8b8c
--- /dev/null
+++ b/bob/ip/binseg/configs/datasets/stare/ah_768.py
@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# coding=utf-8
+
+"""STARE dataset for Vessel Segmentation (annotator AH)
+
+Configuration resolution: 768 x 768 (after padding and resizing)
+
+"""
+from bob.ip.binseg.configs.datasets.stare import _maker_square
+
+dataset = _maker_square("ah")
diff --git a/setup.py b/setup.py
index 56f99b6..50310bf 100644
--- a/setup.py
+++ b/setup.py
@@ -60,6 +60,7 @@ setup(
             "csv-dataset-example = bob.ip.binseg.configs.datasets.csv",
             # drive dataset - retinography
             "drive = bob.ip.binseg.configs.datasets.drive.default",
+            "drive-768 = bob.ip.binseg.configs.datasets.drive.default_768",
             "drive-2nd = bob.ip.binseg.configs.datasets.drive.second_annotator",
             "drive-xtest = bob.ip.binseg.configs.datasets.drive.xtest",
             "drive-mtest = bob.ip.binseg.configs.datasets.drive.mtest",
@@ -69,6 +70,7 @@ setup(
             "drhagis = bob.ip.binseg.configs.datasets.drhagis.default",
             # stare dataset - retinography
             "stare = bob.ip.binseg.configs.datasets.stare.ah",
+            "stare-768 = bob.ip.binseg.configs.datasets.stare.ah_768",
             "stare-2nd = bob.ip.binseg.configs.datasets.stare.vk",
             "stare-xtest = bob.ip.binseg.configs.datasets.stare.xtest",
             "stare-mtest = bob.ip.binseg.configs.datasets.stare.mtest",
@@ -76,13 +78,16 @@ setup(
             "stare-ssl = bob.ip.binseg.configs.datasets.stare.ssl",
             # iostar - retinography
             "iostar-vessel = bob.ip.binseg.configs.datasets.iostar.vessel",
+            "iostar-vessel-768 = bob.ip.binseg.configs.datasets.iostar.vessel_768",
             "iostar-vessel-xtest = bob.ip.binseg.configs.datasets.iostar.vessel_xtest",
             "iostar-vessel-mtest = bob.ip.binseg.configs.datasets.iostar.vessel_mtest",
             "iostar-disc = bob.ip.binseg.configs.datasets.iostar.optic_disc",
+            "iostar-disc-512 = bob.ip.binseg.configs.datasets.iostar.optic_disc_512",
             "iostar-vessel-covd = bob.ip.binseg.configs.datasets.iostar.covd",
             "iostar-vessel-ssl = bob.ip.binseg.configs.datasets.iostar.ssl",
             # hrf - retinography
             "hrf = bob.ip.binseg.configs.datasets.hrf.default",
+            "hrf-768 = bob.ip.binseg.configs.datasets.hrf.default_768",
             "hrf-xtest = bob.ip.binseg.configs.datasets.hrf.xtest",
             "hrf-mtest = bob.ip.binseg.configs.datasets.hrf.mtest",
             "hrf-highres = bob.ip.binseg.configs.datasets.hrf.default_fullres",
@@ -90,6 +95,7 @@ setup(
             "hrf-ssl = bob.ip.binseg.configs.datasets.hrf.ssl",
             # chase-db1 - retinography
             "chasedb1 = bob.ip.binseg.configs.datasets.chasedb1.first_annotator",
+            "chasedb1-768 = bob.ip.binseg.configs.datasets.chasedb1.first_annotator_768",
             "chasedb1-2nd = bob.ip.binseg.configs.datasets.chasedb1.second_annotator",
             "chasedb1-xtest = bob.ip.binseg.configs.datasets.chasedb1.xtest",
             "chasedb1-mtest = bob.ip.binseg.configs.datasets.chasedb1.mtest",
@@ -97,18 +103,26 @@ setup(
             "chasedb1-ssl = bob.ip.binseg.configs.datasets.chasedb1.ssl",
             # drionsdb - retinography
             "drionsdb = bob.ip.binseg.configs.datasets.drionsdb.expert1",
+            "drionsdb-512 = bob.ip.binseg.configs.datasets.drionsdb.expert1_512",
             "drionsdb-2nd = bob.ip.binseg.configs.datasets.drionsdb.expert2",
+            "drionsdb-2nd-512 = bob.ip.binseg.configs.datasets.drionsdb.expert2_512",
             # drishti-gs1 - retinography
             "drishtigs1-disc = bob.ip.binseg.configs.datasets.drishtigs1.disc_all",
+            "drishtigs1-disc-512 = bob.ip.binseg.configs.datasets.drishtigs1.disc_all_512",
             "drishtigs1-cup = bob.ip.binseg.configs.datasets.drishtigs1.cup_all",
+            "drishtigs1-cup-512 = bob.ip.binseg.configs.datasets.drishtigs1.cup_all_512",
             "drishtigs1-disc-any = bob.ip.binseg.configs.datasets.drishtigs1.disc_any",
             "drishtigs1-cup-any = bob.ip.binseg.configs.datasets.drishtigs1.cup_any",
             # refuge - retinography
             "refuge-cup = bob.ip.binseg.configs.datasets.refuge.cup",
+            "refuge-cup-512 = bob.ip.binseg.configs.datasets.refuge.cup_512",
             "refuge-disc = bob.ip.binseg.configs.datasets.refuge.disc",
+            "refuge-disc-512 = bob.ip.binseg.configs.datasets.refuge.disc_512",
             # rim one r3 - retinography
             "rimoner3-cup = bob.ip.binseg.configs.datasets.rimoner3.cup_exp1",
             "rimoner3-disc = bob.ip.binseg.configs.datasets.rimoner3.disc_exp1",
+            "rimoner3-cup-512 = bob.ip.binseg.configs.datasets.rimoner3.cup_exp1_512",
+            "rimoner3-disc-512 = bob.ip.binseg.configs.datasets.rimoner3.disc_exp1_512",
             "rimoner3-cup-2nd = bob.ip.binseg.configs.datasets.rimoner3.cup_exp2",
             "rimoner3-disc-2nd = bob.ip.binseg.configs.datasets.rimoner3.disc_exp2",
             # montgomery county - cxr
-- 
GitLab


From 0a94aafeddffb9bf8fb51992a0cf255c66de91a6 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 19 Nov 2021 18:04:32 +0100
Subject: [PATCH 2/2] [conda] Pin matplotlib to x.x

---
 conda/meta.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/conda/meta.yaml b/conda/meta.yaml
index e008029..7632495 100644
--- a/conda/meta.yaml
+++ b/conda/meta.yaml
@@ -46,7 +46,7 @@ requirements:
     - {{ pin_compatible('scipy') }}
     - {{ pin_compatible('pytorch', max_pin="x.x") }}
     - {{ pin_compatible('torchvision', max_pin="x.x") }}
-    - {{ pin_compatible('matplotlib') }}
+    - {{ pin_compatible('matplotlib', max_pin="x.x") }}
     - {{ pin_compatible('pandas') }}
     - {{ pin_compatible('pillow') }}
     - {{ pin_compatible('psutil') }}
-- 
GitLab