Keras簡單使用在keras中建立模型測試自己的圖片一些有用的函數(持續更新)

Keras簡單使用

在keras中建立模型

相對於自己寫機器學習相關的函數，keras更能快速搭建模型，流程如下：

1. 通過調用下面的函數創建模型

2. 通過調用 model.compile(optimizer = "...", loss = "...", metrics = ["accuracy"])編譯模型

3. 通過調用 model.fit(x = ..., y = ..., epochs = ..., batch_size = ...)在訓練集上訓練模型

4. 通過調用model.evaluate(x = ..., y = ...)在測試集上測試模型

如果你想查閱更多有關model.compile(), model.fit(), model.evaluate() 的信息和它們的參數, 請參考官方文檔 Keras documentation.

代碼如下：

def model(input_shape):
    # Define the input placeholder as a tensor with shape input_shape. Think of this as your input image!
    X_input = Input(input_shape)
​
    # Zero-Padding: pads the border of X_input with zeroes
    X = ZeroPadding2D((3, 3))(X_input)
​
    # CONV -> BN -> RELU Block applied to X
    X = Conv2D(32, (7, 7), strides = (1, 1), name = 'conv0')(X)
    X = BatchNormalization(axis = 3, name = 'bn0')(X)
    X = Activation('relu')(X)
​
    # MAXPOOL
    X = MaxPooling2D((2, 2), name='max_pool')(X)
​
    # FLATTEN X (means convert it to a vector) + FULLYCONNECTED
    X = Flatten()(X)
    X = Dense(1, activation='sigmoid', name='fc')(X)
​
    # Create model. This creates your Keras model instance, you'll use this instance to train/test the model.
    model = Model(inputs = X_input, outputs = X, name='HappyModel')
​
return model

step 1:
happyModel = HappyModel(X_train.shape[1:]) # 只保留一個例子
​
step 2:
happyModel.compile(optimizer = 'sgd', loss = 'binary_crossentropy', metrics = ['accuracy'])
​
step 3：
happyModel.fit(x = X_train,y = Y_train, epochs = 5, batch_size = 16)
​
step 4：
preds =  happyModel.evaluate(x = X_test, y = Y_test)
# preds[0] = Loss
# preds[1] = Test Accuracy

此時，模型參數均已確定，可用來測試自己的圖片

測試自己的圖片

1 img_path = 'your picture path'
2 img = image.load_img(img_path, target_size=(64, 64))
3 imshow(img)
4 ​
5 x = image.img_to_array(img)
6 x = np.expand_dims(x, axis=0)
7 x = preprocess_input(x)
8 ​
9 print(happyModel.predict(x))

一些有用的函數(持續更新)

1. happyModel.summary()：統計並打印如下內容

   

2. plot_model()畫出流程圖

   plot_model(happyModel, to_file='HappyModel.png')
   SVG(model_to_dot(happyModel).create(prog='dot', format='svg'))