如何显示来自 tensorflow_hub 的模型的架构？

Question

我从“https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/1”下载了一个模型如何打印此模型的架构？我想检查这个模型中的网络。此模型如何处理数据。

Answer 1

我自己找到的

func LoadModel(modelPath string, modelNames []string) *tf.SavedModel {
    model, err := tf.LoadSavedModel(modelPath, modelNames, nil) 
    if err != nil {
        log.Fatal("LoadSavedModel(): %v", err)
      }

    log.Println("List possible ops in graphs") 
    for _, op := range model.Graph.Operations() {
        //log.Printf("Op name: %v, on device: %v", op.Name(), op.Device())
        log.Printf("Op name: %v", op.Name())
    }
    return model
}

如何显示来自 tensorflow_hub 的模型的架构？

How to show the architecture of a model from tensorflow_hub?

tensorflow2.0