yan/dddd_trainer
1
0
Fork 0
mirror of https://github.com/netfun2000/dddd_trainer.git synced 2026-08-12 23:59:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

beta0.1

Browse Source
This commit is contained in:
sml2h3
2022-02-22 16:45:43 +08:00
parent 8814267725
commit 608bfec699
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/cache_data.py
+1 -1
View File
@@ -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")