Fork自 OpenDocCN / apachecn-dl-zh
GitCode(gitcode.net)2024年7月9日维护升级公告
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
# 数据集准备 我们的任务是建立一个区分狗和猫的图像分类器。我们从 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`中实现,可以在本书的代码库中下载。