From 98bfed176125043babb07daf649dcc5544abdf5c Mon Sep 17 00:00:00 2001 From: Erin Antono <103962075+eantono-pge@users.noreply.github.com> Date: Thu, 28 Dec 2023 13:57:41 -0800 Subject: [PATCH] Update multi_task.py fixed bug to actually use data_prefix --- mmpretrain/datasets/multi_task.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mmpretrain/datasets/multi_task.py b/mmpretrain/datasets/multi_task.py index 443df0e7d7d..c60b216cf24 100644 --- a/mmpretrain/datasets/multi_task.py +++ b/mmpretrain/datasets/multi_task.py @@ -35,21 +35,21 @@ class MultiTaskDataset: "metainfo": { "tasks": [ - 'gender' + 'gender', 'wear' ] }, "data_list": [ { "img_path": "a.jpg", - gt_label:{ + "gt_label":{ "gender": 0, "wear": [1, 0, 1, 0] } }, { "img_path": "b.jpg", - gt_label:{ + "gt_label":{ "gender": 1, "wear": [1, 0, 1, 0] } @@ -260,7 +260,7 @@ def parse_data_info(self, raw_data): assert 'img_path' in raw_data, \ "The item doesn't have `img_path` field." data = dict( - img_path=self._join_root(raw_data['img_path']), + img_path=self.file_backend.join_path(raw_data['img_path'], self.data_prefix), gt_label=raw_data['gt_label'], ) return data