Python机器学习手册：从数据预处理到深度学习

20.15

　图像分类

｜

333

20.15

　图像分类

问题描述

使用卷积神经网络对一组图像分类。

解决方案

使用

Keras

创建至少有一个卷积层的神经网络 ：

import numpy as np

from keras.datasets import mnist

from keras.models import Sequential

from keras.layers import Dense, Dropout, Flatten

from keras.layers.convolutional import Conv2D, MaxPooling2D

from keras.utils import np_utils

from keras import backend as K

#

设置色彩通道值优先

K.set_image_data_format("channels_first")

#

设置随机种子

np.random.seed(0)

334

｜

第

20

章

神经网络

#

图像信息

channels = 1

height = 28

width = 28

#

从

MNIST

数据集中读取数据和目标

(data_train, target_train), (data_test, target_test) ...