图像数据识别的模型

模型参数设置与模型构建及训练

from keras.models import Sequential
from keras.layers import Dense, Activation
from keras.callbacks import ModelCheckpoint

model = Sequential()
model.add(Dense(units=64, input_dim=100))
model.add(Activation("relu"))
model.add(Dense(units=64, input_dim=100))
model.add(Activation("softmax"))
#完成模型的搭建后,我们需要使用.compile()方法来编译模型:
model.compile(loss='categorical_croosentropy',metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5, batch_size=32)
loss_and_metrics = model.evaluate(x_test, y_test, batch_size=128)
classes = model.predict(x_test, batch_size=128)
model.save('my_model.h5')

#更改loss函数和优化器
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy])

checkpointer = ModelCheckpoint(filepath="checkpoint-{epoch:02d}e-val_acc_{val_acc:2f}.hdf5"
,save_best_only=true, verbose=1, period=50)

model.fit(data,labels, epoch=10,batch_size=32, callbacks=[checkpointer])

#调用Checkpoint保存的model
model = load_model('checkpoint-05e-val_acc_0.58.hdf5')


#模型选取
from keras.application.vgg16 import VGG16
from keras.application.vgg19 import VGG19
from keras.application.inception_v3 import InceptionV3
from keras.application.resnet50 import ResNet50

model_vgg16_conv = VGG16(weights=None, include_top=False, pooling='avg')
output_vgg16_conv = model_vgg16_conv(input)
x = output_vgg16_conv
input = Input(shape=(width,height,channel),name='image_input')
x = Dense(clazz, activation='softmax', name='predictions')(x)

#Create your own model
model = Model(inputs=input, outputs=x)
model.complie(loss=keras.losses.categorical_crossentropy,
optimizer=keras.optimizers.Adam(lr=lr,decay=0),metrics=['acc])

#load all Images
def LoadImageGen(files_data, labels_data,batch=32, label="label"):
      start = 0
      while start < len(file_data):
            stop = start + batch
            if stop > len(files_data):
                  stop = len(file_data)
            imgs = []
            labels = []
            for i in range(start, stop):
                  imgs.append(LoadImage(file_data[i]))
                  labels.append(label_data[i])
            yield(np.array(imgs),np.array(labels))
            if start + batch < len(files_data):
                  start +=batch
            else:
                  zip_data = list(zip(files_data,labels_data))
                  random.shuffle(zip_data)
                  files_data, labels_data = zip(*zip_data)
                  start=0
# load Images to training model
model.fit_generator(
      LoadImageGen(train_x,train_y, batch=batch,label = "train"),
      steps_per_epoch=int(len(train_x)/batch),
      epochs = epoch,
      verbose = 1,
      validation_data = LoadImageGen(test_x,test_y, batch=batch,label = "test"),
      validation_steps = int(len(test_x)/batch),
      callbacks=[
            EarlyStopping(monitor='val_acc',patience=patienceEpoch)),
            modelCheckpoint
            ]
      )

#模型可视化,Tensoborad
#采用keras特有的fit()进行指定callbacks函数即可,代码如下
from keras.callbacks import TensorBoard
from keras.models import Sequential
……
tbCallBack = keras.callbacks.TensorBoard(log_dir='tensorboard',
histogram_freq=1,write_graph=True,write_images=True)
model_history = model.fit(
X_train, y_train,batch_size=batch_size,epochs=epochs,verbose=1,valiation_data=(X_val,y_val),
callbacks = [EarlyStopping(patience=patience,model='min',verbose=1),tbCallBack])

#数据测试:对测试数据集进行验证,并输出测试结果
from keras.models import load_model
model = load_model(*./*.hdf5)

predictLabel=[]
for imgName in os.list(path_base):
      img = LoadImage(os.path.join(path_base, imgName))
      res = np.argmax(model.predict(np.array([img]))
      predictLabel.append(LABELS[res])
acc = round(metrics.precision_score(trueLabel,predictLabel,average='macro'),4)
recall = round(metrics.recall_score(trueLabel,predictLabel,average='macro'),4)
f1_score = round(metrics.f1_score(trueLabel, predictLabel, average='macro'),4)

print("Test acc:{}, Test recall, Test F1_score:{}".format(acc_recall,f1_score))

VGG16:VGG(visual geometry group,超分辨率测试序列)

参考:https://zhuanlan.zhihu.com/p/41423739
共包含13卷积层(Convolutional Layer,表示为conv3-XXXX)+3个连接层(Fully connected Layer,表示为FC-XXXX)+5个池化层(Pool layer,表示maxpool),VGG16的16代表权重系数,maxpool没有权重系数,故16=13+3.

原文地址:https://www.cnblogs.com/TheFaceOfAutumnWhenSummerEnd/p/13880686.html