分类：CNN图像分类器

2021
07-22

基于PyTorch实现一个简单的CNN图像分类器

目录一.加载数据1.继承Dataset类并重写关键方法2.使用Dataloader加载数据二.模型设计三.训练四.测试结语pytorch中文网：https://www.pytorchtutorial.com/pytorch官方文档：https://pytorch.org/docs/stable/index.html一.加载数据Pytorch的数据加载一般是用torch.utils.data.Dataset与torch.utils.data.Dataloader两个类联合进行。我们需要继承Dataset来定义自己的数据集类，然后在训练时用Dataloader加载自定义的数据集类... 继续阅读 >