代码拉取完成,页面将自动刷新
import torch
import torch.nn as nn
class VGG(nn.Module):
def __init__(self, num_classes):
super().__init__()
self.num_classes = num_classes
self.conv1_1 = nn.Conv2d(in_channels=3, out_channels=64,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv1_2 = nn.Conv2d(in_channels=64,out_channels=64,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.pool1 = nn.MaxPool2d(kernel_size=2,stride=2)
self.conv2_1 = nn.Conv2d(in_channels=64,out_channels=128,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv2_2 = nn.Conv2d(in_channels=128,out_channels=128,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.pool2 = nn.MaxPool2d(kernel_size=2,stride=2)
self.conv3_1 = nn.Conv2d(in_channels=128,out_channels=256,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv3_2 = nn.Conv2d(in_channels=256,out_channels=256,padding=1,kernel_size=3,stride=1)
self.pool3 = nn.MaxPool2d(kernel_size=2,stride=2)
self.conv4_1 = nn.Conv2d(in_channels=256,out_channels=512,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv4_2 = nn.Conv2d(in_channels=512,out_channels=512,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv4_2 = nn.Conv2d(in_channels=512,out_channels=512,padding=1,kernel_size=3,stride=1)
self.pool4 = nn.MaxPool2d(kernel_size=2,stride=2)
self.conv5_1 = nn.Conv2d(in_channels=512,out_channels=512,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv5_2 = nn.Conv2d(in_channels=512,out_channels=512,padding=1,kernel_size=3,stride=1)
self.relu = nn.ReLU()
self.conv5_3 = nn.Conv2d(in_channels=512,out_channels=512,padding=1,kernel_size=3,stride=1)
self.pool5 = nn.MaxPool2d(kernel_size=2,stride=2)
self.liner6_1 = nn.Linear(in_features=512*7*7,out_features=4096)
self.relu = nn.ReLU()
self.dropout6_1 = nn.Dropout(p=0.5)
self.liner6_2 = nn.Linear(in_features=4096,out_features=4096)
self.relu = nn.ReLU()
self.liner6_3 = nn.Linear(in_features=4096, out_features=self.num_classes)
self.dropout6_2 = nn.Dropout(p=0.5)
def forward(self,x):
x = self.conv1_1(x)
x = self.relu(x)
x = self.conv1_2(x)
x = self.relu(x)
x = self.pool1(x)
x = self.conv2_1(x)
x = self.relu(x)
x = self.conv2_2(x)
x = self.relu(x)
x = self.pool3(x)
x = self.conv4_1(x)
x = self.relu(x)
x = self.conv4_2(x)
x = self.relu(x)
x = self.pool4(x)
x = self.conv5_1(x)
x = self.relu(x)
x = self.conv5_2(x)
x = self.relu(x)
x = self.conv5_3(x)
x = self.relu(x)
x = self.pool5(x)
x = x.view(x.size(0), -1)
x = self.liner6_1(x)
x = self.relu(x)
x = self.dropout6_1(x)
x = self.liner6_2(x)
x = self.relu(x)
x = self.dropout6_2(x)
x = self.liner6_3(x)
x = self.dropout6_3(x)
return x
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。