《30天吃掉那只 TensorFlow2.0》 1-2 图片数据建模流程范例 (cifar2图片分类问题)

Posted 2022-08-23 风信子的猫Redamancy

1-2 图片数据建模流程范例 (cifar2图片分类问题)

文章目录

• • 1-2 图片数据建模流程范例 (cifar2图片分类问题)
  • • 一，准备数据
    • 二，定义模型
    • 三，训练模型
    • 四，评估模型
    • 五，使用模型
    • 六，保存模型

一，准备数据

cifar2数据集为cifar10数据集的子集，只包括前两种类别airplane和automobile。

训练集有airplane和automobile图片各5000张，测试集有airplane和automobile图片各1000张。

cifar2任务的目标是训练一个模型来对飞机airplane和机动车automobile两种图片进行分类。

我们准备的Cifar2数据集的文件结构如下所示。

<