From 608bfec699446040aea205387c9a2129ea5fa575 Mon Sep 17 00:00:00 2001 From: sml2h3 Date: Tue, 22 Feb 2022 16:45:43 +0800 Subject: [PATCH] beta0.1 --- utils/cache_data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/cache_data.py b/utils/cache_data.py index 474c028..10ba633 100644 --- a/utils/cache_data.py +++ b/utils/cache_data.py @@ -34,7 +34,7 @@ class CacheData: def __get_label_from_name(self, base_path: str): files = os.listdir(base_path) logger.info("\nFiles number is {}.".format(len(files))) - self.__collect_data(files, base_path) + self.__collect_data(files, base_path, []) def __get_label_from_file(self, base_path: str): labels_path = os.path.join(base_path, "labels.txt")