PyTorch中使用迁移训练(Transfer Learning)进行图像分类
PyTorch使用方便,易于学习,开发效率很高。在这篇文章中,我们描述了如何在PyTorch中进行图像分类。我们将使用CalTech256数据集的子集对10种不同种类动物的图像进行分类。我们将介绍数据集准备,数据增强,然后逐步来构建分类器。预训练模型ResNet50已经学习了低层次的图像特征,如边缘、纹理等,我们使用迁移学习来复用这些低层次特征,然后训练我们的分类器来学习我们的数据集中图像中的更高...
原创
2020-02-11 14:20:47 ·
3701 阅读 ·
1 评论