
vgg16_classification_with_pytorch.py
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
这段代码实现了一个使用PyTorch框架基于VGG16网络进行图像分类的任务。通过预训练模型迁移学习完成特定数据集上的分类工作。
在使用PyTorch 1.5实现的VGG16分类模型上,在真实数据集上的测试已经成功完成。
全部评论 (0)
还没有任何评论哟~


简介:
这段代码实现了一个使用PyTorch框架基于VGG16网络进行图像分类的任务。通过预训练模型迁移学习完成特定数据集上的分类工作。
在使用PyTorch 1.5实现的VGG16分类模型上,在真实数据集上的测试已经成功完成。


