From 87e0f42904bba36cfbec537a318c17bdc4a82e01 Mon Sep 17 00:00:00 2001 From: lxs01517514 Date: Mon, 30 Oct 2023 15:17:00 +0800 Subject: [PATCH] update --- .../classification/data_sources/test_cls_caltech_datasource.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_datasets/classification/data_sources/test_cls_caltech_datasource.py b/tests/test_datasets/classification/data_sources/test_cls_caltech_datasource.py index 039083b7..8fd86f91 100644 --- a/tests/test_datasets/classification/data_sources/test_cls_caltech_datasource.py +++ b/tests/test_datasets/classification/data_sources/test_cls_caltech_datasource.py @@ -41,7 +41,7 @@ def test_caltech256(self): results = data_source[idx] img = results['img'] label = results['gt_labels'] - self.assertEqual(img.mode, 'RGB') + self.assertEqual(img.mode, 'L') self.assertIn(label, list(range(len(data_source.CLASSES)))) img.close()