html5国内网站欣赏,商城网站建设二次开发,三星网站建设内容,紫砂壶网站开发与设计报告论文在本篇博客中#xff0c;我们将通过两个主要部分来演示如何使用 PyTorch 实现 ZFNet#xff0c;并在 MNIST 数据集上进行训练和测试。ZFNet#xff08;ZFNet#xff09;是基于卷积神经网络#xff08;CNN#xff09;的图像分类模型#xff0c;广泛用于图像识别任务。
环… 在本篇博客中我们将通过两个主要部分来演示如何使用 PyTorch 实现 ZFNet并在 MNIST 数据集上进行训练和测试。ZFNetZFNet是基于卷积神经网络CNN的图像分类模型广泛用于图像识别任务。
环境准备 在开始之前请确保你的环境已经安装了以下依赖
pip install torch torchvision matplotlib tqdm
一、训练部分训练 ZFNet 模型
首先我们需要准备训练数据、定义 ZFNet 模型并进行模型训练。
1. 数据加载与预处理
MNIST 数据集由 28x28 的手写数字图像组成。我们将通过 torchvision.datasets 来加载数据并进行必要的预处理。
import torch
import torch.optim as optim
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
from zfnet import ZFNet # 假设 ZFNet 定义在 zfnet.py 文件中
from tqdm import tqdm # 导入 tqdm
from torch.cuda.amp import autocast, GradScaler # 导入混合精度训练def prepare_data(batch_size128, num_workers2, data_dirD:/workspace/data):准备 MNIST 数据集并返回数据加载器:param batch_size: 批处理大小:param num_workers: 数据加载的工作线程数:param data_dir: 数据存储的目录:return: 训练数据加载器transform transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5,), (0.5,)) # 正则化])trainset datasets.MNIST(rootdata_dir, trainTrue, downloadTrue, transformtransform)trainloader DataLoader(trainset, batch_sizebatch_size, shuffleTrue, pin_memoryTrue, num_workersnum_workers)return trainloader2. 初始化模型与优化器
在这里我们将初始化模型和优化器。我们选择 Adam 优化器并且为提高计算效率我们采用混合精度训练。
def initialize_device():初始化计算设备GPU 或 CPU:return: 计算设备device torch.device(cuda if torch.cuda.is_available() else cpu)print(fUsing device: {device})return devicedef initialize_model(device):初始化模型并移动到指定设备:param device: 计算设备:return: 初始化好的模型model ZFNet().to(device) # 假设 ZFNet 是自定义模型return modeldef initialize_optimizer(model, lr0.001):初始化优化器:param model: 需要优化的模型:param lr: 学习率:return: 优化器optimizer optim.Adam(model.parameters(), lrlr)return optimizer 3. 训练模型
使用训练数据进行训练并且每训练一个 epoch 就更新一次进度条同时使用混合精度训练来提高效率。
def train_model(model, trainloader, criterion, optimizer, num_epochs5, devicecuda):训练模型:param model: 训练的模型:param trainloader: 数据加载器:param criterion: 损失函数:param optimizer: 优化器:param num_epochs: 训练的轮数:param device: 计算设备scaler GradScaler() # 用于自动缩放梯度for epoch in range(num_epochs):model.train()running_loss 0.0# 使用 tqdm 包裹 DataLoader 来显示进度条with tqdm(trainloader, unitbatch, descfEpoch {epoch 1}/{num_epochs}) as tepoch:for inputs, labels in tepoch:# 直接将数据和标签移动到 GPUinputs, labels inputs.to(device, non_blockingTrue), labels.to(device, non_blockingTrue)optimizer.zero_grad()# 混合精度前向和反向传播with autocast(): # 自动混合精度outputs model(inputs)loss criterion(outputs, labels)# 反向传播与优化scaler.scale(loss).backward() # 使用混合精度反向传播scaler.step(optimizer) # 更新参数scaler.update() # 更新缩放因子running_loss loss.item()# 更新进度条显示tepoch.set_postfix(lossrunning_loss / (tepoch.n 1))# 打印每个 epoch 的平均损失print(fEpoch {epoch 1}, Loss: {running_loss / len(trainloader)})# 保存模型torch.save(model.state_dict(), zfnet_model.pth)print(Model saved as zfnet_model.pth)4. 主函数
在主函数中我们会初始化设备、模型、损失函数并启动训练过程。
if __name__ __main__:主函数组织所有步骤的执行# 数据加载trainloader prepare_data()# 设备选择device initialize_device()# 模型初始化model initialize_model(device)# 损失函数criterion torch.nn.CrossEntropyLoss()# 优化器初始化optimizer initialize_optimizer(model)# 启动训练train_model(model, trainloader, criterion, optimizer, num_epochs5, devicedevice) 二、测试部分评估 ZFNet 模型
训练完成后我们将加载训练好的模型并在测试集上评估其性能。
1. 加载和预处理数据
import torch
import matplotlib.pyplot as plt
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
from zfnet import ZFNet # 假设 ZFNet 定义在 zfnet.py 文件中def load_and_preprocess_data(batch_size1000):加载并预处理 MNIST 数据集:param batch_size: 数据加载的批次大小:return: 测试数据加载器transform transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5,), (0.5,))])# 下载 MNIST 测试集testset datasets.MNIST(rootD:/workspace/data, trainFalse, downloadTrue, transformtransform)# 数据加载器testloader DataLoader(testset, batch_sizebatch_size, shuffleFalse)return testloader 2. 加载训练好的模型
def load_and_preprocess_data(batch_size1000):加载并预处理 MNIST 数据集:param batch_size: 数据加载的批次大小:return: 测试数据加载器transform transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5,), (0.5,))])# 下载 MNIST 测试集testset datasets.MNIST(rootD:/workspace/data, trainFalse, downloadTrue, transformtransform)# 数据加载器testloader DataLoader(testset, batch_sizebatch_size, shuffleFalse)return testloaderdef load_trained_model(model_pathzfnet_model.pth):加载训练好的模型:param model_path: 模型文件路径:return: 加载的模型model ZFNet()model.load_state_dict(torch.load(model_path))model.eval() # 设置为评估模式return model 3. 评估模型
def evaluate_model(model, testloader):评估模型在测试集上的表现:param model: 训练好的模型:param testloader: 测试数据加载器:return: 模型准确率correct 0total 0with torch.no_grad():for inputs, labels in testloader:outputs model(inputs)_, predicted torch.max(outputs, 1)total labels.size(0)correct (predicted labels).sum().item()accuracy correct / totalreturn accuracy 4. 可视化预测结果
def visualize_predictions(model, testloader, num_images6):可视化模型对多张测试图片的预测结果:param model: 训练好的模型:param testloader: 测试数据加载器:param num_images: 显示图像的数量model.eval()data_iter iter(testloader)images, labels next(data_iter)outputs model(images)_, predicted torch.max(outputs, 1)# 绘制结果fig, axes plt.subplots(2, 3, figsize(10, 7))axes axes.ravel()for i in range(num_images):ax axes[i]img images[i].numpy().transpose(1, 2, 0) # 将 Tensor 转换为 NumPy 数组并转置为 HWC 格式ax.imshow(img.squeeze(), cmapgray) # squeeze 去除单通道维度ax.set_title(fPred: {predicted[i].item()} | Actual: {labels[i].item()})ax.axis(off)plt.tight_layout()plt.show()5. 主函数
在测试阶段我们加载模型并在测试数据集上评估它。
def main():主函数组织数据加载、模型加载、评估和可视化步骤# 加载并预处理数据testloader load_and_preprocess_data()# 加载训练好的模型model load_trained_model()# 评估模型accuracy evaluate_model(model, testloader)print(fAccuracy: {accuracy * 100:.2f}%)# 可视化预测结果visualize_predictions(model, testloader, num_images6)if __name__ __main__:main() 结语
通过本文的介绍我们实现了一个基于 ZFNet 模型的图像分类任务使用 PyTorch 对 MNIST 数据集进行训练与测试并展示了如何进行混合精度训练以提高效率。在未来你可以根据不同的任务修改模型结构、优化器或者训练策略进一步提升性能。 完整项目ZFNet-PyTorch: 使用 PyTorch 实现 ZFNet 进行 MNIST 图像分类https://gitee.com/qxdlll/zfnet-py-torch 文章转载自： http://www.morning.pzbqm.cn.gov.cn.pzbqm.cn http://www.morning.dmtld.cn.gov.cn.dmtld.cn http://www.morning.ppzgr.cn.gov.cn.ppzgr.cn http://www.morning.csnch.cn.gov.cn.csnch.cn http://www.morning.wfykn.cn.gov.cn.wfykn.cn http://www.morning.plhhd.cn.gov.cn.plhhd.cn http://www.morning.mljtx.cn.gov.cn.mljtx.cn http://www.morning.prhfc.cn.gov.cn.prhfc.cn http://www.morning.zlrrj.cn.gov.cn.zlrrj.cn http://www.morning.trfh.cn.gov.cn.trfh.cn http://www.morning.qphgp.cn.gov.cn.qphgp.cn http://www.morning.ktdqu.cn.gov.cn.ktdqu.cn http://www.morning.hbpjb.cn.gov.cn.hbpjb.cn http://www.morning.rkgyx.cn.gov.cn.rkgyx.cn http://www.morning.jbgzy.cn.gov.cn.jbgzy.cn http://www.morning.kybjr.cn.gov.cn.kybjr.cn http://www.morning.rlpmy.cn.gov.cn.rlpmy.cn http://www.morning.c7495.cn.gov.cn.c7495.cn http://www.morning.nytpt.cn.gov.cn.nytpt.cn http://www.morning.jcypk.cn.gov.cn.jcypk.cn http://www.morning.bmgdl.cn.gov.cn.bmgdl.cn http://www.morning.fgkwh.cn.gov.cn.fgkwh.cn http://www.morning.yuminfo.com.gov.cn.yuminfo.com http://www.morning.qnxkm.cn.gov.cn.qnxkm.cn http://www.morning.rqmqr.cn.gov.cn.rqmqr.cn http://www.morning.xmyrn.cn.gov.cn.xmyrn.cn http://www.morning.bwhcl.cn.gov.cn.bwhcl.cn http://www.morning.sxwfx.cn.gov.cn.sxwfx.cn http://www.morning.rqckh.cn.gov.cn.rqckh.cn http://www.morning.nlrxh.cn.gov.cn.nlrxh.cn http://www.morning.qxnlc.cn.gov.cn.qxnlc.cn http://www.morning.yesidu.com.gov.cn.yesidu.com http://www.morning.mwkwg.cn.gov.cn.mwkwg.cn http://www.morning.sogou66.cn.gov.cn.sogou66.cn http://www.morning.yjprj.cn.gov.cn.yjprj.cn http://www.morning.lpzqd.cn.gov.cn.lpzqd.cn http://www.morning.ypdhl.cn.gov.cn.ypdhl.cn http://www.morning.ggcjf.cn.gov.cn.ggcjf.cn http://www.morning.lwdzt.cn.gov.cn.lwdzt.cn http://www.morning.hxwrs.cn.gov.cn.hxwrs.cn http://www.morning.lxctl.cn.gov.cn.lxctl.cn http://www.morning.gnghp.cn.gov.cn.gnghp.cn http://www.morning.dywgl.cn.gov.cn.dywgl.cn http://www.morning.gydsg.cn.gov.cn.gydsg.cn http://www.morning.qctsd.cn.gov.cn.qctsd.cn http://www.morning.yfzld.cn.gov.cn.yfzld.cn http://www.morning.fwkq.cn.gov.cn.fwkq.cn http://www.morning.rkrl.cn.gov.cn.rkrl.cn http://www.morning.dnvhfh.cn.gov.cn.dnvhfh.cn http://www.morning.gjqgz.cn.gov.cn.gjqgz.cn http://www.morning.gyqnc.cn.gov.cn.gyqnc.cn http://www.morning.qpqb.cn.gov.cn.qpqb.cn http://www.morning.cjsrg.cn.gov.cn.cjsrg.cn http://www.morning.trzzm.cn.gov.cn.trzzm.cn http://www.morning.nmkbl.cn.gov.cn.nmkbl.cn http://www.morning.pghry.cn.gov.cn.pghry.cn http://www.morning.fdmtr.cn.gov.cn.fdmtr.cn http://www.morning.fnfhs.cn.gov.cn.fnfhs.cn http://www.morning.pzcjq.cn.gov.cn.pzcjq.cn http://www.morning.smygl.cn.gov.cn.smygl.cn http://www.morning.rwyw.cn.gov.cn.rwyw.cn http://www.morning.bsxws.cn.gov.cn.bsxws.cn http://www.morning.rpwck.cn.gov.cn.rpwck.cn http://www.morning.nxbkw.cn.gov.cn.nxbkw.cn http://www.morning.fkyqm.cn.gov.cn.fkyqm.cn http://www.morning.nysjb.cn.gov.cn.nysjb.cn http://www.morning.jfjbl.cn.gov.cn.jfjbl.cn http://www.morning.skrh.cn.gov.cn.skrh.cn http://www.morning.yrck.cn.gov.cn.yrck.cn http://www.morning.rqlbp.cn.gov.cn.rqlbp.cn http://www.morning.rfhwc.cn.gov.cn.rfhwc.cn http://www.morning.tllws.cn.gov.cn.tllws.cn http://www.morning.rmxk.cn.gov.cn.rmxk.cn http://www.morning.zbkdm.cn.gov.cn.zbkdm.cn http://www.morning.hghhy.cn.gov.cn.hghhy.cn http://www.morning.nbnpb.cn.gov.cn.nbnpb.cn http://www.morning.zrjzc.cn.gov.cn.zrjzc.cn http://www.morning.yxnkr.cn.gov.cn.yxnkr.cn http://www.morning.yrnyz.cn.gov.cn.yrnyz.cn http://www.morning.ryztl.cn.gov.cn.ryztl.cn
