src/utils/tools.py
@@ -2,7 +2,7 @@ import numpy as np import time def file_fliter(folder:str, file_ext:str, sort_format): def file_fliter(folder:str, file_ext:str, sort_format = None): files_List = os.listdir(folder) target_files = [] @@ -11,8 +11,10 @@ for file in files_List: if file.split('.')[-1] == file_ext: target_files.append(file) # sort the list by the sort_format target_files.sort(key=sort_format) if sort_format is not None: # sort the list by the sort_format target_files.sort(key=sort_format) return target_files