基于VGG网络的花分类
VGG-使用块的想法首先出现在牛津大学的 视觉几何组(visualgeometry Group (VGG)的 VGG网络 中。通过使用循环和子程序,可以很容易地在任何现代深度学习框架的代码中实现这些重复的结构。
模型
VGG 网络可以分为两部分:第一部分主要由卷积层和池化层组成,第二部分由全连接层组成。
前后有4个不同的网络 可根据cfgs搭建。
import torch.nn as nn
import torch
cfgs = {
'vgg11': [64, 'M', 128, 'M', 256
原创
2021-03-25 13:18:43 ·
773 阅读 ·
0 评论