深度学习笔记09-猫狗识别2(Tensorflow)

m0_67869333

于 2024-12-27 19:34:32 发布

阅读量957

点赞数 14

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签：深度学习笔记 tensorflow

本文链接：https://blog.csdn.net/m0_67869333/article/details/144776174

🍨 本文为🔗365天深度学习训练营中的学习记录博客
🍖 原作者：K同学啊

一、前期工作

1.导入数据并查看

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import os,PIL,pathlib
import warnings
# 支持中文
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号
#隐藏警告
warnings.filterwarnings('ignore')

data_dir = pathlib.Path("./T8")
image_count = len(list(data_dir.glob('*/*')))

print("图片总数为：",image_count)

二、图片预处理

1.加载数据

batch_size = 64
img_height = 224
img_width = 224

"""
关于image_dataset_from_directory()的详细介绍可以参考文章：https://mtyjkh.blog.csdn.net/article/details/117018789
"""
train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    data_dir,
    validation_split=0.2,
    subset="training",
    seed=12,
    image_size=(img_height, img_width),
    batch_size=batch_size)
val_ds = tf.keras.preprocessing.image_dataset_from_directory(
    data_dir,
    validation_split=0.2,
    subset="validation",
    seed=12,
    image_size=(img_height, img_width),
    batch_size=batch_size)

class_names = train_ds.class_names
print(class_names)

2.再一次检查数据

for image_batch, labels_batch in train_ds:
    print(image_batch.shape)
    print(labels_batch.shape)
    break

最低0.47元/天解锁文章

200万优质内容无限畅学