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-21 02:53:47 +08:00
parent 089a9f24dc
commit 8814267725
4 changed files with 19 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/cache_data.py
+4 -4
View File
@@ -55,12 +55,10 @@ class CacheData:
logger.info("\nChecking labels.txt ...")
error_files = set(labels_filename_lines).difference(set(files))
logger.info("\nCheck labels.txt end! {} errors!".format(len(error_files)))
for ef in error_files:
labels_lines.remove(ef)
del files
self.__collect_data(labels_lines, images_path, is_file=True)
self.__collect_data(labels_lines, images_path, error_files, is_file=True)
def __collect_data(self, lines, base_path, is_file=False):
def __collect_data(self, lines, base_path, error_files, is_file=False):
labels = []
caches = []
@@ -72,6 +70,8 @@ class CacheData:
else:
filename = file
label = "_".join(filename.split("_")[:-1])
if filename in error_files:
continue
label = label.replace(" ", "")
if filename.split('.')[-1] in self.allow_ext:
if " " in filename: