使用Pytorch和预训练的VGG16对图片进行分类

1.加载预训练模型 首先要从Pytorch导入VGG,torchvision.models`模块的作用是加载并一些常用的神经网络框架,比如AlexNet、VGG、ResNet、Inception等,包括他们的网络结构以及预训练好的模型。今天我们要使用的是VGG,导入方式如下: ```p......

使用Pytorch展示MNIST数据集中的单张图片

MNIST数据集应该可以说是深度学习中的`Hello World 程序`了,拿他练手不仅可以学习Pytoch等深度学习框架,还可以对深度学习的模型训练流程有一个大体的了解。这篇文章主要是想用Pytorch对MNIST数据集中的单张图片进行展示,代码如下: ```python import t......