PyTorch Transfer Learning Image Classifier:这是Udacity颁发的“使用Python进行AI编程”纳米学位认证的最终

PyTorch转移学习图像分类器 这是Udacity颁发的“使用Python进行AI编程”纳米学位认证的最终项目。 在这个最后的项目中,我构建了一个软件程序,该程序可以使用PyTorch进行神经网络转移学习。 项目详细信息:●用Python和PyTorch编码以生成数据的数据转换。 ●使用PyTorch的torchvision.models(vgg16和densitynet161)进行预先的传输学习,以生成和训练有关数据转换的神经网络。 ●模型在20个纪元内获得了83%的准确度。 编写了保存和重新加载检查点的代码。 ●使用Python的argparse模块构建的命令行应用程序,允许用户加载数据并训练全新的神经网络。 用户可以指定:学习率,隐藏单位,历元,GPU使用率,保存路径以及从PyTorch的torchvision.models中进行选择。