将keras训练的h5模型转成pb格式

骑猪看日落 2022-01-22 01:19 296阅读 0赞

import os  
import tensorflow as tf  
from keras import backend as K  
import logging  
from tensorflow.python.util import compat  
from keras.layers import Conv2D, Dense, Input, add, Activation, AveragePooling2D, GlobalAveragePooling2D  
from keras.models import Model  
os.environ\["CUDA\_VISIBLE\_DEVICES"\] = '0'  
config = tf.ConfigProto()  
config.gpu\_options.allow\_growth = True  
K.set\_session(tf.Session(config=config))  
K.set\_learning\_phase(0)

def export\_savedmodel(model,model\_path='./'):  
logging.info("Model input: \{\}, output: \{\}".format(model.input, model.output))  
model\_signature = tf.saved\_model.signature\_def\_utils.predict\_signature\_def(  
inputs=\{'input': model.input\}, outputs=\{'output': model.output\})

model\_version = 1  
export\_path = os.path.join(  
compat.as\_bytes(model\_path), compat.as\_bytes(str(model\_version)))  
logging.info("Export the model to \{\}".format(export\_path))

builder = tf.saved\_model.builder.SavedModelBuilder(export\_path)  
builder.add\_meta\_graph\_and\_variables(  
sess=K.get\_session(),  
tags=\[tf.saved\_model.tag\_constants.SERVING\],  
clear\_devices=True,  
signature\_def\_map=\{  
tf.saved\_model.signature\_constants.DEFAULT\_SERVING\_SIGNATURE\_DEF\_KEY:  
model\_signature  
\})  
builder.save()  
\#下面是自定义的网络模型  
img\_input = Input(shape=input\_shape)  
output = densenet(img\_input,num\_classes)  
model = Model(img\_input, output)  
model\_path1='./saved\_models/4\_dzj\_model.h5'  
model.load\_weights(model\_path1)  
model\_path = ''  
export\_savedmodel(model, model\_path)

发表评论取消回复

表情：

评论列表（有 0 条评论，296人围观）

还没有评论，来说两句吧...

相关阅读

相关 linux环境将amr格式转成wav格式

windows系统是需要jar包的，下面贴链接可供下载。 /\\ \ 将amr格式转成mp3格式 \ @param amrFilePath amr文件 \ @ret

ゝ一纸荒年。/ 2023年10月17日 21:16/ 0 赞/ 180 阅读

相关 TensorFlow将ckpt模型转为pb模型

-- coding: utf-8 -- import tensorflow as tf from tensorflow.python.framewor

「爱情、让人受尽委屈。」/ 2023年07月09日 06:27/ 0 赞/ 30 阅读

相关转换pb_Keras模型转换成Tensorflow模型方法

![46750eba0315531fd2232ac2cf381157.png][] 由于生产需要、项目组需要，需要将之前神经网络训练的模型（keras模型hdf5类型），转换

我会带着你远行/ 2023年01月07日 10:22/ 0 赞/ 270 阅读

相关 Keras训练的h5文件转pb文件并用Tensorflow加载

一、背景为了快速的搭建神经网络、训练模型，使用了Keras框架来搭建网络并进行训练，得到训练的h5模型文件后，需要将模型部署成服务，而pb格式的文件一般比较适合部署，p

快来打我*/ 2022年12月12日 15:03/ 0 赞/ 813 阅读

相关 Python 转化成 PB 格式数据

一、概述 Protocol Buffers 是 Google 公司开发的一种轻便高效的结构化数据存储格式，可以用于结构化数据串行化，或者说序列化。它很适合做数据存储或 R

野性酷女/ 2022年11月16日 13:59/ 0 赞/ 339 阅读

相关 COCO格式转yolov5训练格式

import json with open('/home/lixuan/fromLX/场景识别/新场景样本/休闲细支新主题陈列/场景识别/训练样本初始

桃扇骨/ 2022年10月08日 15:19/ 0 赞/ 216 阅读

相关将数据转成CSV格式

import csv from distutils.log import warn as printf DATA = ( (9

朴灿烈づ我的快乐病毒、/ 2022年03月30日 11:30/ 0 赞/ 437 阅读

相关将图片转成base64格式

将图片转成base64好处是减少页面的http请求,个人建议用在那种多页面用同一张比较小的图，因为大的图转码生成的字符串编码会很长,写入页面影响页面的可读性，不怎么方面后期的调

偏执的太偏执、/ 2022年03月22日 15:47/ 0 赞/ 437 阅读

相关 keras 预训练模型的使用方法

Tensorflow 几个最新版本的更新大力推崇 Keras 相当于官宣啊相信Keras的可用性给大家做深度学习带来方便畅快的同时逐渐会变成主流的开发组件

喜欢ヅ旅行/ 2022年02月04日 16:06/ 0 赞/ 420 阅读

相关将keras训练的h5模型转成pb格式

import os import tensorflow as tf from keras import backend as K import logging

骑猪看日落/ 2022年01月22日 01:19/ 0 赞/ 297 阅读