From 8116f54a4a93fd1880e85c937677012c6f034b93 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 19 May 2020 19:53:36 +0200
Subject: [PATCH] [configs.datasets.hrf] Add full-resolution train/test sets as
evaluation datasets
---
bob/ip/binseg/configs/datasets/hrf/default.py | 6 +++++
bob/ip/binseg/configs/datasets/hrf/mtest.py | 2 ++
bob/ip/binseg/configs/datasets/hrf/xtest.py | 25 +++++++++++--------
3 files changed, 22 insertions(+), 11 deletions(-)
diff --git a/bob/ip/binseg/configs/datasets/hrf/default.py b/bob/ip/binseg/configs/datasets/hrf/default.py
index 9089e745..3afa4131 100644
--- a/bob/ip/binseg/configs/datasets/hrf/default.py
+++ b/bob/ip/binseg/configs/datasets/hrf/default.py
@@ -9,4 +9,10 @@
"""
from bob.ip.binseg.configs.datasets.hrf import _maker_1168
+
dataset = _maker_1168("default")
+
+from bob.ip.binseg.configs.datasets.hrf.default_fullres import dataset as _fr
+
+dataset["train (full resolution)"] = _fr["train"]
+dataset["test (full resolution)"] = _fr["test"]
diff --git a/bob/ip/binseg/configs/datasets/hrf/mtest.py b/bob/ip/binseg/configs/datasets/hrf/mtest.py
index b73e64b0..e0209439 100644
--- a/bob/ip/binseg/configs/datasets/hrf/mtest.py
+++ b/bob/ip/binseg/configs/datasets/hrf/mtest.py
@@ -12,6 +12,8 @@ from bob.ip.binseg.configs.datasets.hrf.xtest import dataset as _xt
dataset = {
"train": _xt["train"],
"test": _xt["test"],
+ "train (full resolution)": _xt["train (full resolution)"],
+ "test (full resolution)": _xt["test (full resolution)"],
"drive (train)": _xt["drive (train)"].copy(
[Crop(75, 10, 416, 544), Pad((21, 0, 22, 0)), Resize(1168)]
),
diff --git a/bob/ip/binseg/configs/datasets/hrf/xtest.py b/bob/ip/binseg/configs/datasets/hrf/xtest.py
index 277537a2..56af0b8e 100644
--- a/bob/ip/binseg/configs/datasets/hrf/xtest.py
+++ b/bob/ip/binseg/configs/datasets/hrf/xtest.py
@@ -13,14 +13,17 @@ from bob.ip.binseg.configs.datasets.hrf.default import dataset as _hrf
from bob.ip.binseg.configs.datasets.iostar.vessel import dataset as _iostar
dataset = {
- "train": _hrf["train"],
- "test": _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"],
- "iostar (train)": _iostar["train"],
- "iostar (test)": _iostar["test"],
- }
+ "train": _hrf["train"],
+ "test": _hrf["test"],
+ "train (full resolution)": _hrf["train (full resolution)"],
+ "test (full resolution)": _fr["test (full resolution)"],
+ "test": _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"],
+ "iostar (train)": _iostar["train"],
+ "iostar (test)": _iostar["test"],
+}
--
GitLab