Netron神经网络结构可视化只显示权重没有箭头,已解决

Netron神经网络结构可视化只显示权重没有箭头,已解决

好不容易写好了网络,保存成了.h5格式,放进Netron里作图发现我的图黑黑丑丑的而且只显示权重,如下图:
Netron神经网络结构可视化只显示权重没有箭头,已解决_第1张图片
可是看别人画得花花绿绿的还漂亮,为什么到我这里就剩权重了呢?查了一些办法,还废了好大一把劲儿特意把.h5的模型转换成了.onnx格式。结果更是废,造成这种情况的原因,其实不是模型的格式问题,而是因为保存模型时犯了错,我只保存了权重,没有保存完整的网络结构,所以才出现这种情况。

import pandas
pandas.DataFrame(history_callback.history).to_csv("/content/drive/My Drive/AlexNet.csv")  #每一次的准确度
model.save_weights('/content/drive/My Drive/AlexNet.h5')    #最优模型

大家注意看模型保存哪里,我其实保存的是权重,只要把model.save_weights() 改成model.save()就好啦!更改语句如下:

model.save('/content/drive/My Drive/Alex.h5')

然后再用Netron作图就变成花花绿绿可爱的网络结构啦~_~
Netron神经网络结构可视化只显示权重没有箭头,已解决_第2张图片
在这里在附一个Netron的下载链接吧: Netron

你可能感兴趣的:(神经网络结构图,神经网络)