2020
09-29
09-29
将keras的h5模型转换为tensorflow的pb模型操作
背景:目前keras框架使用简单,很容易上手,深得广大算法工程师的喜爱,但是当部署到客户端时,可能会出现各种各样的bug,甚至不支持使用keras,本文来解决的是将keras的h5模型转换为客户端常用的tensorflow的pb模型并使用tensorflow加载pb模型。h5_to_pb.pyfromkeras.modelsimportload_modelimporttensorflowastfimportosimportos.pathasospfromkerasimportbackendasK#路径参数input_path='inputpath'weight_...
继续阅读 >