如何获取项目中所有模型文件的名称(uri)

import os
import glob

def get_model_files(directory):
    model_files = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            # 判断文件类型或文件名是否符合模型文件的命名规则
            if file.endswith('.h5') or file.endswith('.pb') or file.endswith('.onnx'):
                # 获取文件名称（URI）
                file_uri = os.path.join(root, file)
                model_files.append(file_uri)
    return model_files

# 调用函数并传入项目目录
project_directory = '/path/to/project'
model_files = get_model_files(project_directory)

# 打印所有模型文件的名称（URI）
for file in model_files:
    print(file)