keras的plot_model错误三连

from keras.layers import Input, Conv2D, MaxPool2D, Flatten, Dense, Activation
from keras import regularizers
from keras.models import Model
import keras.activations as activation
from keras.utils import plot_model
'''
原味Lenet,没有bn,dropout之类的处理
2020229
chenglong
'''

class Lenet():
    def __init__(self,data_size=(100,32,32,3),
                 n_classes=10):
        # 5
        if len(data_size) == 4:
            self.data_size = data_size
        else:
            raise TypeError
        self.n_classes = n_classes
        self.decay_weight = 0.5e-3
    @property
    def data_shape(self):
        return self.data_size[1:]
    def build(self):
        data_input=Input(batch_shape=self.data_size)
        net =Conv2D(6,(5,5),strides=(1,1),
                    padding="valid",activation="sigmoid",
                    kernel_regularizer=regularizers.l2(self.decay_weight),
                    name="c1")(data_input)
        # net = BatchNormalization()
        net = MaxPool2D((2,2),strides=(2,2),name="s2")(net)
        net = Conv2D(16,(5,5),strides=(1,1),
                     activation="relu",padding="same",
                     kernel_regularizer=regularizers.l2(self.decay_weight),
                     name="c3")(net)
        net = MaxPool2D((2,2),strides=(2,2),name="s4")(net)
        net = Flatten()(net)
        net = Dense(120,name="f5",activation="sigmoid")(net)
        net = Dense(84,activation="sigmoid",name="f6")(net)
        net = Dense(self.n_classes, activation="softmax")(net)
        net = Activation('softmax')(net)
        model = Model(inputs=data_input,outputs=net)
        plot_model(model, to_file=os.path.join('./', "base_model.png"), show_shapes=True)
        model.summary()
        return net


model = Lenet()
model.build()

 1 会报错,如下图。需要需要安装pydot和cython两个包。pip安装就行。

 2 安装完还会报错,如下图,没装Graphviz这个软件。

 去官网,http://www.graphviz.org/,下载。

然后一直next就行了。然后把’D:\Program Files (x86)\Graphviz2.38\bin‘添加到环境变量的path。

3 然后会报这个错误

然后用everything找的pydot.py文件,然后把文件中的self.prog = 'dot'修改为self.prog = 'dot.exe'就行了。

然后运行,这个图不是上面代码的结果。

猜你喜欢

转载自blog.csdn.net/weixin_39875161/article/details/105248438