大家好!这节课我们要用PyTorch打造一个"鸟类观察家兼航空管制员"AI系统,让它能区分天上飞的是鸟还是飞机。我们会用CIFAR-10数据集(已经包含这两类图片),全程笑料不断,保证学得开心!
首先安装必要的库(如果你还没安装的话):
pip install torch torchvision matplotlib
CIFAR-10数据集就像一本包含10类物体的相册,其中正好有鸟(class 2)和飞机(class 0):
import torch
from torchvision import