初识神经网络——基于TensorFlow&Keras完成猫狗数据集

最新推荐文章于 2023-09-18 07:58:37 发布

MrKaj

最新推荐文章于 2023-09-18 07:58:37 发布

阅读量464

点赞数

分类专栏：机器学习从入门到放弃文章标签： python 机器学习

本文链接：https://blog.csdn.net/MrKaj/article/details/118001225

版权

文章目录

一、实验准备
二、开始实验
- - 2.1 制作数据集
  - 2.2 卷积神经网络CNN
三、使用VGG19实现猫狗分类
四、资料引用

一、实验准备

1.1 实验环境：

anaconda3 + python3.8 + Jupyter notebook

1.2 理解卷积神经网络CNN
这部分网上资源很多
1.3 下载安装TensorFlow 和Keras
1)命令行安装
pip install tensorflow
pip install keras
在这里插入图片描述
2)anaconda软件中安装

keras同理

二、开始实验

2.1 制作数据集

从kaggle网站下载数据集
1）图片分类：

import os, shutil
# The path to the directory where the original
# dataset was uncompressed
original_dataset_dir = 'D:\\Anaconda-Project\\CatAndDog\\kaggle_Dog&Cat\\kaggle_Dog&Cat\\train'

# The directory where we will
# store our smaller dataset
base_dir = 'D:\\Anaconda-Project\\CatAndDog\\kaggle_Dog&Cat\\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')
os.mkdir(validation_cats_dir)

# Directory with our validation dog pictures
validation_dogs_dir = os.path.join(validation_dir, 'dogs')
os.mkdir(validation_dogs_dir)

# Directory with our validation cat pictures
test_cats_dir = os.path.join(test_dir, 'cats')
os.mkdir(test_cats_dir)

# Directory with our validation dog pictures
test_dogs_dir = os.path.join(test_dir, 'dogs')
os.mkdir(test_dogs_dir)

# Copy first 1000 cat images to train_cats_dir
fnames = ['cat.{}.jpg'.format(i) for i in range(1000)]
for fname in fnames:
    src = os.path.join(original_dataset_dir, fname)
    dst = os.path.join(train_cats_dir, fname)
    shutil.copyfile(src, dst)

# Copy next 500 cat images to validation_cats_dir
fnames = ['cat.{}.jpg'.format(i) for i in range(1000, 1500)]
for fname in fnames:
    src = os.path.join(original_dataset_dir, fname)
    dst = os.path.join(validation_cats_dir, fname)
    shutil.copyfile(src, dst)
    
# Copy next 500 cat images to test_cats_dir
fnames = ['cat.{}.jpg'.format(i) for i in range(1500, 2000)]
for fname in fnames:
    src = os.path.join(original_dataset_dir, fname)
    dst = os.path.join(test_cats_dir, fname)
    shutil.copyfile(src, dst)
    
# Copy first 1000 dog images to train_dogs_dir
fnames = ['dog.{}.jpg'.format(i) for i in range(1000)]
for fname in fnames:
    src = os.path.join(original_dataset_dir, fname)
    dst = os.path.join(train_dogs_dir, fname)
    shutil.copyfile(src, dst)
    
# Copy next 500 dog images to validation_dogs_dir
fnames = ['dog.{}.jpg'.format(i) for i in range(1000, 1500)]
for fname in fnames:
    src = os.path.join(original_dataset_dir, fname)
    dst = os.path.join(validation_dogs_dir, fname)
    shutil.copyfile(src, dst)
    
# Copy next 500 dog images to test_dogs_dir
fnames = ['dog.{}.jpg'.format(i) for i in range(1500, 2000)]
for fname in fnames:
    src = os.path