from keras import layers
from keras import models
from keras.layers import Dropout
from keras import optimizers
from keras.models import load_model
train_dir = ‘./data/train/’
validation_dir = ‘./data/validation/’
model_file_name = ‘cat_dog_model.h5’
def init_model():
model = models.Sequential()
KERNEL_SIZE = (3, 3)
model.add(layers.Conv2D(filters=32, kernel_size=KERNEL_SIZE, activation=‘relu’, input_shape=(150, 150, 3)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(filters=64, kernel_size=KERNEL_SIZE, activation=‘relu’))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(filters=128, kernel_size=KERNEL_SIZE, activation=‘relu’))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(filters=128, kernel_size=KERNEL_SIZE, activation=‘relu’))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Flatten())
model.add(layers.Dense(512, activation=‘relu’))
model.add(Dropout(0.5))
model.add(layers.Dense(1, activation=‘sigmoid’))
model.compile(loss=‘binary_crossentropy’,
optimizer=optimizers.RMSprop(lr=1e-3),
metrics=[‘accuracy’])
return model
def fig_loss(history):
history_dict = history.history
loss_values = history_dict[‘loss’]
val_loss_values = history_dict[‘val_loss’]
epochs = range(1, len(loss_values) + 1)
plt.plot(epochs, loss_values, ‘b’, label=‘Training loss’)
plt.plot(epochs, val_loss_values, ‘r’, label=‘Validation loss’)
plt.title(‘Training and validation loss’)
plt.xlabel(‘Epochs’)
plt.ylabel(‘Loss’)
plt.legend()
plt.grid()
plt.show()
def fig_acc(history):
history_dict = history.history
acc = history_dict[‘accuracy’]
val_acc = history_dict[‘val_accuracy’]
epochs = range(1, len(acc) + 1)
plt.plot(epochs, acc, ‘g’, label=‘Training acc’)
plt.plot(epochs, val_acc, ‘r’, label=‘Validation acc’)
plt.title(‘Training and validation accuracy’)
plt.xlabel(‘Epochs’)
plt.ylabel(‘Accuracy’)
plt.legend()
plt.grid()
plt.show()
def fit(model):
train_datagen = ImageDataGenerator(rescale=1. / 255)
validation_datagen = ImageDataGenerator(rescale=1. / 255)
train_generator = train_datagen.flow_from_directory(
train_dir,
target_size=(150, 150),
batch_size=256,
class_mode=‘binary’)
validation_generator = validation_datagen.flow_from_directory(
validation_dir,
target_size=(150, 150),
batch_size=64,
class_mode=‘binary’)
history = model.fit_generator(
train_generator,
steps_per_epoch=,
epochs=10,
validation_data=validation_generator,
validation_steps=,
)
model.save(model_file_name)
fig_loss(history)
fig_acc(history)
def predict():
model = load_model(model_file_name)
print(model.summary())
img_path = ‘./data/test/cat/cat.4021.jpg’
img = image.load_img(img_path, target_size=(150, 150))
img_tensor = image.img_to_array(img)
img_tensor = img_tensor / 255
img_tensor = np.expand_dims(img_tensor, axis=0)
其形状为 (1, 150, 150, 3)
plt.imshow(img_tensor[0])
plt.show()
result = model.predict(img_tensor)
print(result)
画出count个预测结果和图像
def fig_predict_result(model, count):
test_datagen = ImageDataGenerator(rescale=1. / 255)
test_generator = test_datagen.flow_from_directory(
‘./data/test/’,
target_size=(150, 150),
batch_size=256,
class_mode=‘binary’)
text_labels = []
plt.figure(figsize=(30, 20))
迭代器可以迭代很多条数据,但我这里只取第一个结果看看
for batch, label in test_generator:
pred = model.predict(batch)
for i in range(count):
true_reuslt = label[i]
最后附上:我们之前因为秋招收集的二十套一二线互联网公司Android面试真题 (含BAT、小米、华为、美团、滴滴)和我自己整理Android复习笔记(包含Android基础知识点、Android扩展知识点、Android源码解析、设计模式汇总、Gradle知识点、常见算法题汇总)
面试成功其实是必然的,因为我做足了充分的准备工作,包括刷题啊,看一些Android核心的知识点,看一些面试的博客吸取大家面试的一些经验,下面这份PDF是我翻阅了差不多1个月左右一些Android大博主的博客从他们那里取其精华去其糟泊所整理出来的一些Android的核心知识点, 全部都是精华中的精华,我能面试到现在资深开发人员跟我整理的这本Android核心知识点有密不可分的关系,在这里本着共赢的心态分享给各位朋友。
这份PDF囊括了JVM,Java集合,Java多线程并发,Java基础,生命周期,微服务, 进程,Parcelable 接口,IPC,屏幕适配,线程异步,ART,架构,Jetpack,NDK开发,计算机网络基础,类加载器,Android 开源库源码分析,设计模式汇总,Gradle 知识点汇总…
由于篇幅有限,就不做过多的介绍,大家请自行脑补
《Android学习笔记总结+移动架构视频+大厂面试真题+项目实战源码》,点击传送门,即可获取!
447)]
这份PDF囊括了JVM,Java集合,Java多线程并发,Java基础,生命周期,微服务, 进程,Parcelable 接口,IPC,屏幕适配,线程异步,ART,架构,Jetpack,NDK开发,计算机网络基础,类加载器,Android 开源库源码分析,设计模式汇总,Gradle 知识点汇总…
由于篇幅有限,就不做过多的介绍,大家请自行脑补
《Android学习笔记总结+移动架构视频+大厂面试真题+项目实战源码》,点击传送门,即可获取!