在公园里，三岁孩童能轻松分辨迎面跑来的是小狗还是猫咪，但对计算机而言，这曾是需要顶尖科学家团队攻克的难题。2012年，深度学习在ImageNet竞赛中一战成名，开启了人工智能的"视觉革命"。如今，借助JupyterLab和TensorFlow框架，即使是编程新手也能在笔记本电脑上构建出媲美专业团队的图像识别系统。本文将带你走进深度学习的世界，手把手教你搭建一个能分辨猫狗的智能系统。

AI开发需要专业工具链，腾讯云HAI非常的方便快捷，内置JupyterLab作为操作台，这个基于网页的交互式开发环境，能实时显示代码运行结果，特别适合机器学习实验。通过简单的pip命令安装TensorFlow、Matplotlib等工具包，就像为实验室购置了试管和培养皿。

pip install tensorflow jupyterlab matplotlib pillow

任何智能系统的培养都始于数据准备。建立清晰的目录结构是成功的第一步：

data/
    train/
        cats/   # 存放1000张猫咪图片
        dogs/   # 存放1000张狗狗图片
    validation/
        cats/   # 存放400张验证用猫咪
        dogs/   # 存放400张验证用狗狗

这模拟了人类的学习过程：先通过大量示例学习（训练集），再通过测试检验学习成果（验证集）。注意保持猫狗图片数量平衡，就像老师不会只教认猫不教认狗。

直接使用原始图片就像只给学生看正脸证件照，现实中对象会有各种角度变化。ImageDataGenerator就是这个问题的解决方案：这相当于给每张图片生成无数个"分身"，让模型见识各种可能的猫咪形态：倒立的、半遮挡的、侧身的...经过这样的训练，模型就像见过世面的宠物医生，能从容应对各种情况。

train_datagen = ImageDataGenerator(
    rotation_range=40,       # 随机旋转±40度
    width_shift_range=0.2,   # 水平平移20%
    height_shift_range=0.2,  # 垂直平移20%
    shear_range=0.2,         # 剪切变换
    zoom_range=0.2,          # 随机缩放
    horizontal_flip=True     # 水平翻转
)

2.2 卷积神经网络：视觉密码破解器

我们构建的CNN（卷积神经网络）就像精密的视觉处理流水线：这个过程就像刑侦专家破案：先寻找蛛丝马迹（边缘检测），然后拼凑线索（特征组合），最后进行综合推理。Dropout层的加入，相当于故意隐藏部分线索，防止模型形成刻板印象。

model = models.Sequential([
    layers.Conv2D(32, (3,3),  # 第一层"显微镜"提取边缘特征
    layers.MaxPooling2D(2,2),  # 浓缩重要信息
    layers.Conv2D(64, (3,3)), # 识别眼睛、耳朵等部件
    layers.MaxPooling2D(2,2),
    layers.Conv2D(128, (3,3)),# 组合部件识别整体
    layers.Flatten(),          # 将立体特征展开
    layers.Dense(512),         # 全连接层综合判断
    layers.Dropout(0.5),       # 防止死记硬背
    layers.Dense(1, activation='sigmoid')  # 输出判断结果
])

设置优化器就像选择教学方法：Adam优化器结合了动量法和自适应学习率的优点，像因材施教的好老师。学习率1e-4的设置需要平衡：太大会错过细节，太小学习效率低下。

model.compile(
    optimizer=tf.keras.optimizers.Adam(learning_rate=1e-4),
    loss='binary_crossentropy',
    metrics=['accuracy']
)

启动训练后，每个epoch都是学习周期：

history = model.fit(
    train_generator,
    steps_per_epoch=100,
    epochs=30,
    validation_data=val_generator
)

观察训练曲线就像查看成绩单：理想状态是训练准确率与验证准确率同步上升，两者差距不大。如果出现"课堂学霸，考试崩溃"的情况（过拟合），就需要增加Dropout比例或收集更多数据。

用Matplotlib绘制的学习曲线，是诊断模型健康的重要工具：两条曲线的趋同程度反映模型的泛化能力。若验证准确率持续低于训练准确率，说明模型产生了"偏科"现象。

plt.figure(figsize=(12, 4))
plt.subplot(1, 2, 1)
plt.plot(acc, label='课堂表现')
plt.plot(val_acc, label='考试成绩')
plt.title('准确率进化史')

训练完成的模型可以保存为.h5文件，就像毕业证书：

加载模型进行预测时，注意保持与训练时相同的预处理流程：

def predict_image(img_path):
    img = image.load_img(img_path, target_size=(150,150))
    x = image.img_to_array(img)/255.
    return 'Dog' if model.predict(x[np.newaxis,...])[0] > 0.5 else 'Cat'

通过这个项目，我们构建了一个猫狗识别器，亲历了深度学习的基本方法论。从数据准备到模型调优的每个环节，都体现着AI哲学。当在JupyterLab中看到第一个成功预测结果时，那闪烁的光标背后，正是一个新智能生命的觉醒。这只是一个起点，调整网络深度、尝试迁移学习、探索目标检测...深度学习的海洋，正在等待你的探索。 Let`s Go~~~

import tensorflow as tf
from tensorflow.keras import layers, models
from tensorflow.keras.preprocessing.image import ImageDataGenerator
import matplotlib.pyplot as plt

train_datagen = ImageDataGenerator(
    rescale=1./255,
    rotation_range=40,
    width_shift_range=0.2,
    height_shift_range=0.2,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True
)

val_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_datagen.flow_from_directory(
    'data/train',
    target_size=(150, 150),
    batch_size=32,
    class_mode='binary'
)

val_generator = val_datagen.flow_from_directory(
    'data/validation',
    target_size=(150, 150),
    batch_size=32,
    class_mode='binary'
)
model = models.Sequential([
    layers.Conv2D(32, (3,3), activation='relu', input_shape=(150,150,3)),
    layers.MaxPooling2D(2,2),
    layers.Conv2D(64, (3,3), activation='relu'),
    layers.MaxPooling2D(2,2),
    layers.Conv2D(128, (3,3), activation='relu'),
    layers.MaxPooling2D(2,2),
    layers.Flatten(),
    layers.Dense(512, activation='relu'),
    layers.Dropout(0.5),
    layers.Dense(1, activation='sigmoid')
])

model.compile(
    loss='binary_crossentropy',
    optimizer=tf.keras.optimizers.Adam(learning_rate=1e-4),
    metrics=['accuracy']
)
history = model.fit(
    train_generator,
    steps_per_epoch=100,  # 根据样本数调整
    epochs=30,
    validation_data=val_generator,
    validation_steps=50
)
acc = history.history['accuracy']
val_acc = history.history['val_accuracy']
loss = history.history['loss']
val_loss = history.history['val_loss']

plt.figure(figsize=(12, 4))
plt.subplot(1, 2, 1)
plt.plot(acc, label='Training Accuracy')
plt.plot(val_acc, label='Validation Accuracy')
plt.legend()
plt.title('Accuracy Curves')

plt.subplot(1, 2, 2)
plt.plot(loss, label='Training Loss')
plt.plot(val_loss, label='Validation Loss')
plt.legend()
plt.title('Loss Curves')
plt.show()
model.save('cat_dog_cnn.h5')

从零开始，用JupyterLab和TensorFlow打造你的第一个猫狗识别模型

当计算机学会"看"世界在公园里，三岁孩童能轻松分辨迎面跑来的是小狗还是猫咪，但对计算机而言，这曾是需要顶尖科学家团队攻克的难题。2012年，深度学习在ImageNet竞赛中一战成名，开启了人工智能的"视觉革命"。如今，借助JupyterLab和TensorFlow框架，即使是编程新手也能在笔记本电脑上构建出媲美专业团队的图像识别系统。本文将带你走进深度学习的世界，手把手教你搭建一个能分辨猫狗的智能

产品KOL

人工智能

客户端

卷积神经网络

自适应学习

图像识别

机器学习

深度学习

精选特惠 拼团嗨购

ai-class

imagerecognition

33 天实现自己的 AI 进化论？发文瓜分机械键盘、耳机等万元奖池！点击参加活动：<a href="https://cloud.tencent.com/developer/article/2503022" target="_blank">https://cloud.tencent.com/developer/article/2503022</a><br/> 
 <img src="https://qcloudimg.tencent-cloud.cn/raw/01543e0407db10898a7ee9f420794432.jpg"/>

技术创作特训营有奖征文

文章

问答

视频

学习中心

腾讯云实验室

直播

竞赛

腾讯云架构师技术同盟交流圈

腾讯云数据库专区

腾讯云原生专区

腾讯混元专区

腾讯云TCE专区

腾讯云Lighthouse专区

腾讯云HAI专区

腾讯云Edgeone专区

腾讯云存储专区

腾讯云智能专区

腾讯轻联专区 

腾讯云开发专区

TAPD专区

腾讯云最具价值专家

腾讯云架构师技术同盟

腾讯云创作之星

腾讯云开发者先锋 

腾讯云AI代码助手

CODING DevOps

Cloud Studio

SDK中心

API中心

命令行工具

文章&问答评论现已支持表情

全新交互，全新视觉，新增快捷键、悬浮工具栏、高亮块等功能并同时优化现有功能，全面提升创作效率和体验

社区富文本编辑器全新改版！诚邀体验～ 

在公园里，三岁孩童能轻松分辨迎面跑来的是小狗还是猫咪，但对计算机而言，这曾是需要顶尖科学家团队攻克的难题。2012年，深度学习在ImageNet竞赛中一战成名，开启了人工智能的"视觉革命"。如今，借助JupyterLab和TensorFlow框架，即使是编程新手也能在笔记本电脑上构建出媲美专业团队的图像识别系统。本文将...

从零开始，用JupyterLab和TensorFlow打造你的第一个猫狗识别模型

从零开始，用JupyterLab和TensorFlow打造你的第一个猫狗识别模型

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐