From e87058e0d88fc219a1a6b7fe491a2852fba5f9d5 Mon Sep 17 00:00:00 2001 From: Timo Tjaden Stomberg <timo.stomberg@uni-bonn.de> Date: Mon, 26 Sep 2022 16:35:07 +0200 Subject: [PATCH] images dataset returns str(file) --- tlib/ttorch/data/images.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tlib/ttorch/data/images.py b/tlib/ttorch/data/images.py index c63f2ac..9c92e69 100644 --- a/tlib/ttorch/data/images.py +++ b/tlib/ttorch/data/images.py @@ -187,7 +187,7 @@ class Dataset(data.basic.Dataset): y = self.labels[index] if self.labels is not None else None - return {'x': x, 'y': y, 'file': file} + return {'x': x, 'y': y, 'file': str(file)} def transform(self, x, y): """ -- GitLab