【文件属性】：
文件名称：pytorch-image-classification:使用pytorch进行图像分类
文件大小：51.39MB
文件格式：ZIP
更新时间：2021-05-29 10:38:53
deep-learning image-classification convolutional-neural-networks transfer-learning pytorch-tutorial
Pytorch-图像分类
使用pytorch进行图像分类的简单演示。 在这里，我们使用包含43956 张图像的自定义数据集，属于11 个类别进行训练（和验证）。 此外，我们比较了三种不同的训练方法。 从头开始培训，微调的convnet和convnet为特征提取，用预训练pytorch模型的帮助。 使用的模型包括： VGG11、Resnet18 和 MobilenetV2 。
依赖关系
Python3，Scikit学习
Pytorch, PIL
Torchsummary，Tensorboard
pip install torchsummary # keras-summary
pip install tensorboard # tensoflow-logging
注意：在训练之前将库更新到最新版本。
怎么跑
下载并提取训练数据集：
运行以下脚本进行训练和/或测试
python t