第四次作业：猫狗大战挑战赛

下载数据：

 

 

数据处理：

 

 

 

 

 

显示图片：

 

 

创建vgg模型：

使用预训练好的 VGG 模型。下载了 ImageNet 的 JSON 文件展示 VGG 模型对本数据的预测结果。在这部分代码中，对输入的5个图片利用VGG模型进行预测。使用softmax对结果进行处理。

 

 

 

修改最后一层，冻结前面层的参数

输出VGG模型结构并设置参数，将Linear层由1000类，替换为2类。并把requires_grad参数设置为False，这样就不保存梯度，在反向传播时就不会去更新权重。训练时，只会更新最后一层的参数。将新的模型载入到GPU中。

 

 

训练并测试：

 

 

 

 

 

 

 

 

 

 

 

 

提交结果：