ModelArts模型管理导入FrozenGraph格式模型

当前ModelArts文档中只有SavedModel格式的案例

ModelArts模型导入事实上也支持使用FrozenGraph进行模型导入。

通过阅读文档https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0093.html

从文档中可以看到有如下方法进行重写

那么这两种模型导入的差别在哪里呢。

如果导入的模型文件是SavedModel，那么一般情况下是不需要 实现__init__(self,model_name,model_path)方法。

但是如果导入的模型是FrozenGraph就需要实现__init__(self,model_name,model_path)方法，并且在此方法中实现模型的加载。

以tensorflow2.0为例

https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0301.html

__init__(self,model_name,model_path)方法中调用了load_model，因此只要在load_model()方法中实现加载FozenGraph格式的模型就可以了。