环境: Pytorch1.1,Python3.6,win10/ubuntu18,GPU 正文 Pytorch构建ResNet18模型并训练,进行真实图片分类: 利用预训练的ResNet18模型进行Fine tune,直接进行图片分类:站在巨人的肩膀上,使用已经在ImageNet上训练好的模型,除了最后一层全连接层,中间层的参数全部迁移到目标模型上,如下图所示 项目结构如下所示 pokemon里面存放数据,分别是五个文件夹,其中每个文件夹分别存放一定数量的图片,总共1000多张图片: best.m