中药材分类实验中要求:调用MobileNet_v2模型,并与ResNet18模型结果比较,结合实验给出实现以上要求的代码

时间: 2024-03-03 18:50:08 浏览: 18
以下是基于PyTorch实现的中药材分类实验,包括调用MobileNet_v2模型并与ResNet18模型结果比较的代码: ```python import torch import torch.nn as nn import torch.optim as optim import torchvision.models as models import torchvision.transforms as transforms import torchvision.datasets as datasets # 定义数据预处理 transform_train = transforms.Compose([ transforms.RandomResizedCrop(224), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) transform_test = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) # 加载数据集 trainset = datasets.ImageFolder('data/train', transform=transform_train) trainloader = torch.utils.data.DataLoader( trainset, batch_size=32, shuffle=True, num_workers=4) testset = datasets.ImageFolder('data/test', transform=transform_test) testloader = torch.utils.data.DataLoader( testset, batch_size=32, shuffle=False, num_workers=4) # 定义模型 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.model = models.resnet18(pretrained=True) self.model.fc = nn.Linear(512, 10) def forward(self, x): return self.model(x) # 定义损失函数和优化器 net_resnet = Net() net_mobilenet = models.mobilenet_v2(pretrained=True) net_mobilenet.classifier[1] = nn.Linear(1280, 10) criterion = nn.CrossEntropyLoss() optimizer_resnet = optim.SGD(net_resnet.parameters(), lr=0.001, momentum=0.9) optimizer_mobilenet = optim.SGD(net_mobilenet.parameters(), lr=0.001, momentum=0.9) # 训练模型 def train(net, optimizer, epoch): net.train() for batch_idx, (data, target) in enumerate(trainloader): optimizer.zero_grad() output = net(data) loss = criterion(output, target) loss.backward() optimizer.step() if batch_idx % 100 == 0: print('Train Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.6f}'.format( epoch, batch_idx * len(data), len(trainloader.dataset), 100. * batch_idx / len(trainloader), loss.item())) def test(net): net.eval() test_loss = 0 correct = 0 with torch.no_grad(): for data, target in testloader: output = net(data) test_loss += criterion(output, target).item() pred = output.max(1, keepdim=True)[1] correct += pred.eq(target.view_as(pred)).sum().item() test_loss /= len(testloader.dataset) print('\nTest set: Average loss: {:.4f}, Accuracy: {}/{} ({:.0f}%)\n'.format( test_loss, correct, len(testloader.dataset), 100. * correct / len(testloader.dataset))) for epoch in range(1, 11): train(net_resnet, optimizer_resnet, epoch) test(net_resnet) for epoch in range(1, 11): train(net_mobilenet, optimizer_mobilenet, epoch) test(net_mobilenet) ``` 在以上代码中,我们首先定义了数据预处理的方法,然后加载了训练集和测试集。接着定义了一个使用ResNet18模型的分类器Net,并使用pretrained参数加载了预训练权重。然后定义了损失函数和优化器,分别使用SGD算法进行优化。接下来训练模型,并在每个epoch结束后使用测试集测试模型的性能。最后,我们定义了另一个使用MobileNet_v2模型的分类器,并重复了训练和测试的过程。通过比较两个模型在测试集上的准确率,我们可以得出它们的性能差异。

相关推荐

最新推荐

recommend-type

使用Keras预训练模型ResNet50进行图像分类方式

主要介绍了使用Keras预训练模型ResNet50进行图像分类方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
recommend-type

Pytorch修改ResNet模型全连接层进行直接训练实例

在本篇文章里小编给大家整理的是关于Pytorch修改ResNet模型全连接层进行直接训练相关知识点,有需要的朋友们参考下。
recommend-type

Tensorflow 2.1训练 实战 cifar10 完整代码 准确率 88.6% 模型 Resnet SENet Inception

模型: Resnet:把前一层的数据直接加到下一层里。减少数据在传播过程中过多的丢失。 SENet: 学习每一层的通道之间的关系 Inception: 每一层都用不同的核(1×1,3×3,5×5)来学习.防止因为过小的核或者过大的核...
recommend-type

【深度学习入门】Paddle实现人脸检测和表情识别(基于TinyYOLO和ResNet18)

Paddle实现人脸检测和表情识别(基于YOLO和ResNet18)一、先看效果:训练及测试结果:UI 界面及其可视化:二、AI Studio 简介:平台简介:创建项目:三、创建AI Studio项目:创建并启动环境:下载数据:下载预训练...
recommend-type

RTL8188FU-Linux-v5.7.4.2-36687.20200602.tar(20765).gz

REALTEK 8188FTV 8188eus 8188etv linux驱动程序稳定版本, 支持AP,STA 以及AP+STA 共存模式。 稳定支持linux4.0以上内核。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

系统函数是1+5*z^(-1)+5*z^(-2)+z^(-3) ,给出Matlab中求该系统频率响应的代码

假设系统函数为H(z),则其频率响应为H(w),可以通过以下代码求解: ``` syms z w H = 1 + 5*z^(-1) + 5*z^(-2) + z^(-3); % 定义系统函数 Hw = subs(H, z, exp(1i*w)); % 将z用e^(jw)代替 Hw = simplify(Hw); % 化简 absHw = abs(Hw); % 求幅度响应 angleHw = angle(Hw); % 求相位响应 ``` 其中,`simplify`函数用于化简表达式,`abs`函数用于求绝对值,`angle`函数用于求相位。
recommend-type

c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf

校园超市商品信息管理系统课程设计旨在帮助学生深入理解程序设计的基础知识,同时锻炼他们的实际操作能力。通过设计和实现一个校园超市商品信息管理系统,学生掌握了如何利用计算机科学与技术知识解决实际问题的能力。在课程设计过程中,学生需要对超市商品和销售员的关系进行有效管理,使系统功能更全面、实用,从而提高用户体验和便利性。 学生在课程设计过程中展现了积极的学习态度和纪律,没有缺勤情况,演示过程流畅且作品具有很强的使用价值。设计报告完整详细,展现了对问题的深入思考和解决能力。在答辩环节中,学生能够自信地回答问题,展示出扎实的专业知识和逻辑思维能力。教师对学生的表现予以肯定,认为学生在课程设计中表现出色,值得称赞。 整个课程设计过程包括平时成绩、报告成绩和演示与答辩成绩三个部分,其中平时表现占比20%,报告成绩占比40%,演示与答辩成绩占比40%。通过这三个部分的综合评定,最终为学生总成绩提供参考。总评分以百分制计算,全面评估学生在课程设计中的各项表现,最终为学生提供综合评价和反馈意见。 通过校园超市商品信息管理系统课程设计,学生不仅提升了对程序设计基础知识的理解与应用能力,同时也增强了团队协作和沟通能力。这一过程旨在培养学生综合运用技术解决问题的能力,为其未来的专业发展打下坚实基础。学生在进行校园超市商品信息管理系统课程设计过程中,不仅获得了理论知识的提升,同时也锻炼了实践能力和创新思维,为其未来的职业发展奠定了坚实基础。 校园超市商品信息管理系统课程设计的目的在于促进学生对程序设计基础知识的深入理解与掌握,同时培养学生解决实际问题的能力。通过对系统功能和用户需求的全面考量,学生设计了一个实用、高效的校园超市商品信息管理系统,为用户提供了更便捷、更高效的管理和使用体验。 综上所述,校园超市商品信息管理系统课程设计是一项旨在提升学生综合能力和实践技能的重要教学活动。通过此次设计,学生不仅深化了对程序设计基础知识的理解,还培养了解决实际问题的能力和团队合作精神。这一过程将为学生未来的专业发展提供坚实基础,使其在实际工作中能够胜任更多挑战。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依