经典卷积模型回顾15—Googlenet实现图像分类(Tensorflow2.0,猫狗分类)
创始人
2024-06-01 06:19:03
0

GoogLeNet是Google在2014年提出的一个深度学习模型,也是当时ImageNet图像分类挑战赛(ILSVRC14)的获胜者,比起先前的模型,GoogLeNet在模型深度和模型参数上都有很大的优化,同时也提高了模型的准确率。GoogLeNet拥有22层网络,其中有9个Inception模块。Inception模块采用1x1、3x3和5x5的卷积核来分别捕捉不同尺度的特征,同时使用池化操作,最后将不同尺度的特征图在深度维度上进行拼接。这样可以保证网络既有足够深度来捕捉复杂特征,又避免了参数过多的问题,同时还能保证计算效率。此外,GoogLeNet还引入了辅助分类器(Auxiliary Classifier)来帮助网络更加快速地收敛。辅助分类器是对中间层的输出进行分类,这些分类器的误差也参与整个网络的反向传播,从而更加有效地更新网络参数,使得网络更加容易收敛。

我们将使用“Dogs vs Cats”数据集进行训练和评估。

首先,我们需要导入必要的库:

``` python

import tensorflow as tf

import matplotlib.pyplot as plt

import numpy as np

import os

import random

from tensorflow.keras.preprocessing.image import ImageDataGenerator

```

接下来,我们将定义一些超参数和数据集路径:

``` python

# 超参数

IMG_HEIGHT = 224

IMG_WIDTH = 224

BATCH_SIZE = 32

EPOCHS = 50

NUM_CLASSES = 2

# 数据集路径

train_dir = './dogs-vs-cats/train'

test_dir = './dogs-vs-cats/test1'

```

然后,我们将创建一个数据增强器,并将其应用于训练数据。 这可以帮助我们避免过度拟合,并提高模型的泛化能力。

``` python

# 数据增强器

train_image_generator = ImageDataGenerator(

    rescale=1./255,

    rotation_range=30,

    width_shift_range=0.2,

    height_shift_range=0.2,

    shear_range=0.2,

    zoom_range=0.2,

    horizontal_flip=True,

    fill_mode='nearest'

)

# 训练数据生成器

train_data_gen = train_image_generator.flow_from_directory(

    train_dir,

    target_size=(IMG_HEIGHT, IMG_WIDTH),

    batch_size=BATCH_SIZE,

    class_mode='categorical'

)

```

接下来,我们将创建一个用于测试数据的数据生成器。

``` python

# 测试数据生成器

test_image_generator = ImageDataGenerator(rescale=1./255)

test_data_gen = test_image_generator.flow_from_directory(

    test_dir,

    target_size=(IMG_HEIGHT, IMG_WIDTH),

    batch_size=BATCH_SIZE,

    class_mode='categorical'

)

```

现在,我们将定义一个GoogleNet模型。我们将使用预训练的模型,并在模型的顶部添加一些全连接层和Dropout层。

``` python

# 定义模型

base_model = tf.keras.applications.InceptionV3(include_top=False, weights='imagenet', input_shape=(IMG_HEIGHT, IMG_WIDTH, 3))

x = base_model.output

x = tf.keras.layers.GlobalAveragePooling2D()(x)

x = tf.keras.layers.Dense(512, activation='relu')(x)

x = tf.keras.layers.Dropout(0.5)(x)

predictions = tf.keras.layers.Dense(NUM_CLASSES, activation='softmax')(x)

model = tf.keras.models.Model(inputs=base_model.input, outputs=predictions)

for layer in base_model.layers:

    layer.trainable = False

```

现在,我们将编译模型并开始训练。

``` python

# 编译模型

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

# 训练模型

history = model.fit(

    train_data_gen,

    epochs=EPOCHS,

    validation_data=test_data_gen

)

```

最后,我们将绘制模型的训练和验证损失和准确率。

``` python

# 绘制损失和准确率

acc = history.history['accuracy']

val_acc = history.history['val_accuracy']

loss = history.history['loss']

val_loss = history.history['val_loss']

epochs_range = range(EPOCHS)

plt.figure(figsize=(8, 8))

plt.subplot(2, 1, 1)

plt.plot(epochs_range, acc, label='Training Accuracy')

plt.plot(epochs_range, val_acc, label='Validation Accuracy')

plt.legend(loc='lower right')

plt.title('Training and Validation Accuracy')

plt.subplot(2, 1, 2)

plt.plot(epochs_range, loss, label='Training Loss')

plt.plot(epochs_range, val_loss, label='Validation Loss')

plt.legend(loc='upper right')

plt.title('Training and Validation Loss')

plt.show()

```

相关内容

热门资讯

开晨会需要说什么内容 开晨会需要说什么内容晨会说的内容可以很多呀,可以说接下来的工作安排,也可以说,前期工作中的一些总结,...
有谁知道在鬼吹灯有声小说中前面... 有谁知道在鬼吹灯有声小说中前面有一段小女孩的声音念:人点烛 鬼吹灯... 这个版本的故事是谁讲的?周...
小蒂皮出自哪本书 小蒂皮出自哪本书小蒂皮出自哪本书我的野生动物朋友(蒂皮·本杰明·奥康迪·德格雷著图书)《我的野生动物...
盘古开天辟地的故事 盘古开天辟地的故事很久很久以前,天和地还没有分开,宇宙混沌一片。有个叫盘古的巨人一直睡在这混沌之中。...
英雄联盟中 赏金猎人怎么样? ... 英雄联盟中 赏金猎人怎么样? 多少钱 金币3150的ADC。入手的话不会吃亏,基本不会压箱底,很热门...
寻找一部累死何以笙箫默之类的小... 寻找一部累死何以笙箫默之类的小说也是辛夷坞的,叫山月不知心底事那就它的作者顾漫的其他小说呗 微微一...
大家谈谈对洪荒小说之中圣人的看... 大家谈谈对洪荒小说之中圣人的看法圣人不死,大盗不止都是人 人有的情绪他们都有 就是拳头大些对...
单位体检,自己一个人去害怕撞到... 单位体检,自己一个人去害怕撞到同事怎么办,不想别人看到观察我,漏出我不成熟样子,显出我的丑态。这让单...
鱼在天上飞? 鱼在天上飞?“鱼在天上飞,鸟在水里游”这句话的意思是鸟儿在水中的倒影就像是在水里游,天空在水中的倒影...
我想复婚,前夫说以后再,我改变... 我想复婚,前夫说以后再,我改变好了,他有可能追我,现在不想复婚,他有喜欢的女人了,我们就不可能了吗感...
原神草神什么时候复刻 原神草神什么时候复刻原神草神复刻是在3.6版本以后。草神纳西妲首次登场是逗腊3.2版本,新角色想要等...
网络拽姐语录小学生? 网络拽姐语录小学生?无论谁离开了你,请你别忘了,他没来之前,你本就是一个人生活。
有一部小说,女主小时候被男主收... 有一部小说,女主小时候被男主收养了,女主叫男主哥哥,男主对女主的我也在找这个小说我之前也看了,好像叫...
高层建筑立面大面积粉刷需要设伸... 高层建筑立面大面积粉刷需要设伸缩缝吗?高层建筑立面大面积粉刷需要设伸缩缝需要的,无正举论地面或立念清...
低头思故乡,为什么把杯思在长,... 低头思故乡,为什么把杯思在长,情愿何处在方歌曲叫什么名字“李白的歌”离开了家乡 背着沉沉的行囊开始了...
李白的诗有多少 李白的诗有多少典故中大多说是三千至四千首,现存不足一千首很多,总之不少~李白一生留下了九百六十多首诗...
怎么才可以学会拿得起放得下……... 怎么才可以学会拿得起放得下…… 有一段感情,正整3年了,可是我还是觉得分手那天就好像昨天发生的一样替...
为什么官方直播间买手机送耳机 为什么官方直播间买手机送耳机吸引消费者,增加销售量。官方直播间作为官方的直播平台,买手机送耳机是为了...
自考通还是一考通好?该选哪个? 自考通还是一考通好?该选哪个?我现在自考行政管理专科。专业课程我应该选自考通好还是一考通好呢?这个很...
淘宝实际付款价格为什么与订单价... 淘宝实际付款价格为什么与订单价格不符是这样的,我买了一双鞋,当时限价160我拍了下来,当时没付款,后...