Ubuntu下面的Keras可视化+权重维度获取-Netron的安装使用

环境:

Ubuntu	19.10
Python	3.6.10
Keras	2.3.1

#---------------------------------------------------------------------------------------------------------------------------------------

操作过程如下:

$ snap install netron

$ netron
Serving at http://localhost:8080

然后在浏览器打开http://localhost:8080

在Open Model上传自己的.h5模型即可。

#---------------------------------------------------------------------------------------------------------------------------------------

准备好数据集:

tsocks wget https://s3.amazonaws.com/img-datasets/mnist.npz

mv mnist.npz ~/.keras/datasets

#---------------------------------------------------------------------------------------------------------------------------------------

运行如下代码(来自[1]):

from keras.models import Model
from keras.layers import Input, Dense
from keras.datasets import mnist
from keras.utils import np_utils(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train=x_train.reshape(x_train.shape[0],-1)/255.0
x_test=x_test.reshape(x_test.shape[0],-1)/255.0
y_train=np_utils.to_categorical(y_train,num_classes=10)
y_test=np_utils.to_categorical(y_test,num_classes=10)inputs = Input(shape=(784, ))
x = Dense(64, activation='relu')(inputs)
x = Dense(64, activation='relu')(x)
y = Dense(10, activation='softmax')(x)model = Model(inputs=inputs, outputs=y)model.save('m1.h5')
model.summary()
model.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['accuracy'])
model.fit(x_train, y_train, batch_size=32, epochs=10)
#loss,accuracy=model.evaluate(x_test,y_test)model.save('m2.h5')
model.save_weights('m3.h5')

运行后我们得到m1,m2,m3三个文件

文件名	保存的函数	保存了图结构	保存了模型参数	可否用Neutron打开
m1.h5	save()	√	X	√
m2.h5	save()	√	√	√
m3.h5	save_weights()	X	√	√

m1.h5打开结果

m2.h5打开结果

m3.h5打开结果

从上面可以看到权重的维度。

所以如[1]所说，没啥事儿的话，尽量使用save()而不是save_weights()

#---------------------------------------------------------------------------------------------------------------------------------------

Reference:

[1]keras保存模型中的save()和save_weights()

Ubuntu下面的Keras可视化+权重维度获取-Netron的安装使用相关推荐

ubuntu下docker环境、php环境以及laravel的安装
2019独角兽企业重金招聘Python工程师标准>>> ubuntu下docker环境.php环境以及laravel的安装因为在学习laravel,需要搭建一个php7的开发环境, ...
ubuntu下QQ, 微信, 讯雷, 网盘等无痛安装, 解决高分辨率问题hidpi
ubuntu下QQ, 微信, 讯雷, 网盘等无痛安装, 解决高分辨率问题hidpi 感谢配置deepin-wine 配置方法下载wine的容器高分辨率问题前提告知解决高分辨率问题,开启Win ...
dpkg命令用法、Ubuntu下deb包的解压、打包、安装、卸载及常用命令参数
dpkg命令的用法不带图简装:https://blog.csdn.net/wanghuohuo13/article/details/78916821?ops_request_misc=&re ...
【GPU+ubuntu】Ubuntu下的Tensorflow-gpu等各种环境和包的安装
配置: CPU:i7 GPU:1070 CUDA:8.0 Cudnn:8.0v6.0 Tensorflow:1.4 双系统:win7 + ubuntu16.04 一.系统安装 Win7 U盘安装Ubu ...
ubuntu下面的git服务器搭建
1.安装相应的软件和依赖 ubuntu:~$ sudo apt-get install git-core openssh-server openssh-client git-core是git版本控制核 ...
js如何获取div下面的input_原生 js 如何获取宽高
foreword(前言) 首先,非常抱歉太长时间没有更新专栏了,这两三个月一直在处理人生两大重要的事...闲话少说,进入正题. 最近项目中需要获取 dom 元素的尺寸,所以借此对比分析一下 js 所有 ...
ubuntu下mongodb及其可视化工具robomongo安装
小编是个纯算法工程师,问我算法可以,别让我整别的东西.前段时间实验室师哥任务交接,将大量的数据存到了mongodb中,对于这个词,小编只听过没见过.然后师哥说了一个可视化的工具robomongo,这个 ...
Ubuntu 下面的aMule 的 Kad 或者 Ed2k 连不上
电驴--电骡爱好者的家.在eMule(电骡)运行过程中需要使用到很多文件,它们将在eMule第一次运行时创建.它们中的大部分会被创建到eMule的config文件夹里,其它的文件会直接存放在eMule ...
ubuntu下面的java一键运行脚本
java的运行分为编译+执行两部分. 怪麻烦的,所以写个脚本一键运行.如下: ------------------------------------------------------------- ...

Ubuntu下面的Keras可视化+权重维度获取-Netron的安装使用

Ubuntu下面的Keras可视化+权重维度获取-Netron的安装使用相关推荐

最新文章

热门文章