ch04s05.md 775 字节
Newer Older
W
wizardforcel 已提交
1 2 3 4 5 6 7 8 9
# 数据集准备

我们的任务是建立一个区分狗和猫的图像分类器。我们从 Kaggle 那里得到一些帮助,[我们可以从中轻松下载数据集](https://www.kaggle.com/c/dogs-vs-cats/data)

在此数据集中,训练集包含 20,000 个标记图像,测试和验证集包含 2,500 个图像。

要使用数据集,必须将每个图像重新整形为 227×227×3。为此,您可以使用`prep_images.py`中的 Python 代码。否则,您可以使用本书仓库中的`trainDir.rar``testDir.rar`文件。它们包含 6,000 个用于训练的犬和猫的重塑图像,以及 100 个重新成形的图像用于测试。

以下部分中描述的以下微调实现在`alexnet_finetune.py`中实现,可以在本书的代码库中下载。