卷积神经网络图像分类

最新推荐文章于 2022-10-19 15:02:59 发布

无由无

最新推荐文章于 2022-10-19 15:02:59 发布

阅读量612

点赞数 4

文章标签：神经网络卷积

本文链接：https://blog.csdn.net/weixin_45134710/article/details/117295896

版权

本文介绍了如何配置TensorFlow和Keras进行深度学习，特别是使用卷积神经网络（CNN）对猫狗图片进行分类。首先，讲解了安装Anaconda和启动Jupyter的过程，接着详细描述了数据集的制作，包括获取猫狗数据集并进行分类。在数据预处理之后，尝试搭建CNN模型，但由于遇到问题导致作业终止。

摘要由CSDN通过智能技术生成

一、配置TensorFlow、Keras

安装Anaconda
可参考前面的博客
启动jupyter
选择Teminal
输入命令行

pip install tensorflow
pip install keras

在这里插入图片描述
等待安装完成即可

二、猫狗数据集

1.制作数据集

获取猫狗数据集
运行程序：根据命名对图片分类

import os, shutil
# The path to the directory where the original
# dataset was uncompressed
original_dataset_dir = 'D:/boot/kaggle_Dog&Cat/train'

# The directory where we will
# store our smaller dataset
base_dir = 'D:/boot/kaggle_Dog&Cat/find_cats_and_dogs'
os.mkdir(base_dir)

# Directories for our training,
# validation and test splits
train_dir = os.path.join(base_dir, 'train')
os.mkdir(train_dir)
validation_dir = os.path.join(base_dir, 'validation')
os.mkdir(validation_dir)
test_dir = os.path.join(base_dir, 'test')
os.mkdir(test_dir)

# Directory with our training cat pictures
train_cats_dir = os.path.join(train_dir, 'cats')
os.mkdir(train_cats_dir)

# Directory with our training dog pictures
train_dogs_dir = os.path.join(train_dir, 'dogs')
os.mkdir(train_dogs_dir)

# Directory with our validation cat pictures
validation_cats_dir = os.path.join(validation_dir, 'cats'