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