猫狗大战——VGG

最新推荐文章于 2024-03-19 21:09:16 发布

Nicolas XL

最新推荐文章于 2024-03-19 21:09:16 发布

阅读量337

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/weixin_44182138/article/details/109746140

版权

本文介绍了在猫狗分类任务中，通过调整VGG模型参数，创建并训练了五个不同版本的模型。最终选择模型三进行测试，经过谷歌云盘数据处理和Colab训练，结果显示原始VGG模型表现优秀，但进一步优化需要更精细的调参。

摘要由CSDN通过智能技术生成

首先修改参数，获得以下几个模型

模型一、

model_vgg_new.classifier._modules['3'] = nn.Linear(4096, 128)
model_vgg_new.classifier._modules['6'] = nn.Linear(128, 2)
model_vgg_new.classifier._modules['7'] = torch.nn.LogSoftmax(dim = 1)

	acc
training	0.4467
testing	0.4165

训练准确度
测试准确度
模型二、

model_vgg_new.classifier._modules['3'] = nn.Linear(4096, 1024)
model_vgg_new.classifier._modules['6'] = nn.Linear(1024, 2)
model_vgg_new.classifier._modules['7'] = torch.nn.LogSoftmax(dim = 1)

	acc
training	0.4833
testing	0.5850

训练正确率
测试正确率
模型三、

model_vgg_new.classifier._modules['3'] = nn.Linear(4096, 4096)
model_vgg_new.classifier._modules['6'] = nn.Linear(4096, 2)
model_vgg_new.classifier._modules['7'] = torch.nn.LogSoftmax(dim = 1)

	acc
training	0.6528
testing	0.9225

训练准确率
测试准确率

模型四、

model_vgg_new.classifier._modules['0'] = nn.Linear(25088, 8192)
model_vgg_new.classifier._modules['3'] = nn.Linear(8192, 8192)
model_vgg_new.classifier._modules['6'] = nn.Linear(8192, 2)
model_vgg_new.classifier._modules['7'] = torch.nn.LogSoftmax(dim = 1)

	acc
training	0.5567
testing	0.6465

训练准确率
测试准确率
模型五、

model_vgg_new.classifier._modules['6'] = nn.Linear(4096, 4096)
model_vgg_new.classifier._modules['7'] = nn.ReLU(inplace=True)
model_vgg_new.classifier._modules['8'] = nn.Dropout(p