From f53de2be3c2d472d4da7b516941f96fdc96de915 Mon Sep 17 00:00:00 2001
From: Claudio Michaelis <claudio.michaelis@uni-tuebingen.de>
Date: Tue, 1 Oct 2019 22:28:03 -0400
Subject: [PATCH] Fix robustness data pipeline (#1476)
---
tools/test_robustness.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tools/test_robustness.py b/tools/test_robustness.py
index e263215..c0489f3 100644
--- a/tools/test_robustness.py
+++ b/tools/test_robustness.py
@@ -350,9 +350,9 @@ def main():
aggregated_results[corruptions[0]][0]
continue
+ test_data_cfg = copy.deepcopy(cfg.data.test)
# assign corruption and severity
if corruption_severity > 0:
- test_data_cfg = copy.deepcopy(cfg.data.test)
corruption_trans = dict(
type='Corrupt',
corruption=corruption,
@@ -368,7 +368,7 @@ def main():
# build the dataloader
# TODO: support multiple images per gpu
# (only minor changes are needed)
- dataset = build_dataset(cfg.data.test)
+ dataset = build_dataset(test_data_cfg)
data_loader = build_dataloader(
dataset,
imgs_per_gpu=1,
--
GitLab