基于Pytorch的VGG实现
今天看了关于VGG的论文,大致看了下他的网络架构,稍微改改实现了下
ABCDE这是VGG不同的网络架构图,其中D就是著名的VGG16其实差别并不是很大,我实现的是最为简单的A网络也就是VGG11,8层卷积与3层全连接层,中间还有一些最大池和rule激活
// An highlighted block
class Net(nn.Module):
def __init__(self):
super(Net,self).__init__()
self.conv1=nn.Conv2d(3,64,kernel_size=3,padding=1)
self.conv2=nn.Conv2d(64,128,kernel_size=3,padding=1)
self.conv3_1=nn.Conv2d(128,256,kernel_size