image_classifier_pytorch:udacity_intro_to_ml_with_pytorch_projcet_2 源码

使用PyTorch的图像分类器 使用PyTorch ND进行ML入门的Udacity项目2 项目描述 在该项目中,我使用深度学习训练图像分类器,以使用vg16_nb预训练模型对花朵的图像进行分类。 模式应用程序之一可能是您将智能手机指向一朵花,而电话应用程序会告诉您它是什么花。但是,可以在您选择的任何数据集上对模型进行重新训练。您可以学习它来识别汽车,将图片对准汽车,然后让应用程序告诉您它的品牌和型号。 用法 下载102个花卉类别的 运行`train.py'训练模型 基本用法: python train.py data_directory 选项: 设置目录以保存检查点: python train.py data_dir --save_dir save_directory 选择架构: python train.py data_dir --arch "vgg13" 设置超参数: python