深度学习-图像识别

图片剪裁模块：ImageDataGenerator

#数据生成器，指定scaling 范围0-1
train_datagen=ImageDataGenerator(rescale=1/255)

train_generator=train_datagen.flow_from_directory(
    './horse-or-human/',
    target_size=(300,300),
    batch_size=32,
    class_mode='binary'
)

优化模型板块：hyperband，hyperparameters

import tensorflow as tf
from kerastuner.tuners import Hyperband
from kerastuner.engine.hyperparameters import HyperParameters

hp=HyperParameters()

def build_model(hp):
    model=keras.models.Sequential()
    model.add(tf.keras.layer.Conv2D(hp.Choice('num_filters_layer0',values=[16,64],default=16,activation='relu'))
    model.add(tf.keras.layers.MaxPooling2D(2,2))

    #用于优化调参
    for I in range(hp.Int('num_filters_layer0',1,3)):
        model.add(tf.keras.layer.Conv2D(hp.Choice('num_filters_layer0',values=[16,64],default=16,activation='relu'))
        model.add(tf.keras.layers.MaxPooling2D(2,2))
        
    model.add(tf.keras.layers.Flatten())
    model.add(tf.keras.layers.Dense(128,activation=tf.nn.relu))
    model.add(tf.keras.layers.Dense(10,activation=tf.nn.softmax))
    model.compile(loss='binary_crossentropy',optimizer=RMSprop(lr=0.001),metrics=['acc'])
    return model

tuner=Hyperband(
    build_model,
    objective='val_acc',
    max_epochs=15,
    directory='horse-or-human',
    hyperparameters=hp,
    project_name='horse_human_project'
)

tuner.search(train_generator,epochs=15,validation_data=xxx)