pytorch
文章平均质量分 54
西西先生666
这个作者很懒,什么都没留下…
展开
-
pytorch实现:Resnet模型识别花朵数据集(参考pytorch官网代码)
代码参考pytorch官网代码写法,包含模型保存、模型加载原创 2022-04-28 17:10:51 · 5525 阅读 · 3 评论 -
pytorch之常用函数整理
pytorch之常用函数整理一、图像预处理函数1.1 torchvision.datasets.ImageFolder()函数二、参数优化函数torch.optim.lr_scheduler.StepLR()函数一、图像预处理函数1.1 torchvision.datasets.ImageFolder()函数ImageFolder函数假设所有的文件按文件夹保存,每个文件夹下存储同一类别的图片,文件夹名为类别名;ImageFolder(root, transform=None, target_tr原创 2022-04-27 11:43:52 · 5365 阅读 · 5 评论 -
pytorch实现:对预训练好的卷积网络微调
pytorch实现:对预训练好的卷积网络微调一、导入所需的包二、微调预训练的VGG16网络三、准备新网络需要的数据四、微调网络的训练和预测由于卷积神经网络层数多,训练参数较多,导致从0开始训练很深的卷积网络非常困难,所以我们可以对预训练好的模型参数进行微调,快速实现自己的任务。pytorch提供ImageNet数据集与预训练好的流行的深度学习网络。本文使用VGG16网络,对网络结构和参数进行微调。数据集来自kaggle:https://www.kaggle.com/slothkong/10-monkey原创 2021-05-02 20:59:04 · 1249 阅读 · 1 评论 -
pytorch实现:卷积神经网络识别MNIST
pytorch实现:卷积神经网络识别MNIST# Author: Liuxin# Time: 2022/4/21# 导入训练所需包import torchimport torch.nn as nnimport torch.optim as optimfrom torchvision import datasets, transformsimport torch.nn.functional as Fimport matplotlib.pyplot as pltimport pandas a原创 2022-04-22 12:07:26 · 2056 阅读 · 0 评论