软件工程第四次作业
软件工程第四次作业
1、迁移学习
下载数据，包括训练集和测试集，训练集包含1800张图(猫的图片900张，狗的图片900张)，测试集包含2000张图。
datasets 是 torchvision 中的一个包，用来加载图像数据。并将图片整理成 $224\times 224 \times 3$ 的大小，进行归一化处理。
使用预训练好的 VGG 模型
设置 required_grad=False，那么，反向传播训练梯度时，前面层的权重就不会自动更新了，训练时只更新最后一层的参数。
2、猫狗大战挑战赛
手动上传猫狗大战测试集，共2000张图片，上传后文件目录如下图所示
通过代码解压test文件，并更换目录
加载测试数据集
将图片序号和结果写入.csv文件中，并手动按照序号升序排序
提交结果，在未对训练模型做出改动的情况下，评测得分为96.15
