MrSong007的博客

什么是PytorchPytorch是一个基于python的库，用它可以方便地构建深度学习模型并应用在不同的领域。但是它不仅仅是另一个深度学习库那么简单，他也是一个科学计算软件包。作为一款基于python的科学计算软件包，主要有两大特色：代替Numpy使用GPUs的能力；一个拥有很大灵活度和速度的深度学习研究平台；Pytorch使用 Tensor 作为他的核心数据结构，这一结构与Nump...

在这一篇文章中，我们描述了如何在 pytorch中进行图像分类。我们将使用Caltech256数据集的一个子集对10种不同动物的图像进行分类。借此来介绍数据集的准备、数据扩充，以及构建分类器的完整步骤。使用转移学习来使用低层次的图像特征，如边缘、纹理等。通过借助一个预训练的模型 resnet50，训练自己的分类器来学习数据集图像中更高级的细节，比如眼睛、腿等。ResNet50 已经在ImageNe...