wu_xinjun
2022-03-15 4033aede52133f254f320838d2339ca3bfcb77e9
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