图像分类是计算机视觉中最基础也是最经典的任务之一。今天我们将带你实战体验如何使用 PyTorch 和 ResNet 构建一个猫狗识别系统。从数据预处理、模型构建、训练调优到模型保存和预测,每一步都细致讲解,带你快速上手!
你可以从 Kaggle 下载经典的猫狗数据集(Dogs vs. Cats):
链接:https://www.kaggle.com/c/dogs-vs-cats/data
将数据集解压后,按如下方式组织数据:
dataset/
├── train/
│ ├── cat.0.jpg
│ ├── dog.0.jpg
│ └── ...
├── val/
│ ├──