如何将CoreML模型中的输出名称从Identity更改为其他名称？

import CoreML

// 1. 加载CoreML模型
guard let modelURL = Bundle.main.url(forResource: "YourModel", withExtension: "mlmodelc") else {
    fatalError("Failed to locate the CoreML model")
}
guard let model = try? MLModel(contentsOf: modelURL) else {
    fatalError("Failed to load the CoreML model")
}

// 2. 查看模型结构
let modelDescription = model.modelDescription

// 3. 修改输出名称
guard var outputDescriptions = modelDescription.outputDescriptionsByName else {
    fatalError("Failed to retrieve output descriptions")
}

// 将输出名称从Identity更改为其他名称
if let identityOutputDescription = outputDescriptions["Identity"], let newOutputDescription = try? identityOutputDescription.rename(name: "NewOutputName") {
    outputDescriptions["NewOutputName"] = newOutputDescription
    outputDescriptions["Identity"] = nil
}

modelDescription.outputDescriptionsByName = outputDescriptions

// 4. 保存修改后的模型
guard let updatedModel = try? MLModel(mlModel: model, modelDescription: modelDescription) else {
    fatalError("Failed to create the updated CoreML model")
}

let updatedModelURL = URL(fileURLWithPath: "/path/to/updatedModel.mlmodelc")
try? updatedModel.write(to: updatedModelURL)