- 博客(2)
- 收藏
- 关注
RSS订阅原创 基于Pytorch和Vgg16实现图片分类
最近在基于Pytorch框架补一些CNN的基础知识，学会自己写简单的卷积神经网络，从加载数据集到训练模型、测试模型、保存模型和输出测试结果，现在来总结一下。首先基于Pytorch实现Vgg16网络，命名为model.py（可为其他任意名字，但是后续导入时要记得更改）import torchimport torch.nn as nnclass VGG16(nn.Module): def __init__(self): super(VGG16, self).__init_
2022-05-26 17:06:35
4653
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人