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'就行了。
然后运行,这个图不是上面代码的结果。