代碼地址：https://github.com/lshiwjx/2s-AGCN
本篇解析代碼為[2s-AGCN/model/aagcn.py]

import mathimport numpy as np import torch import torch.nn as nn from torch.autograd import Variabledef import_class(name):components = name.split('.')mod = __import__(components[0])for comp in components[1:]:mod = getattr(mod, comp)return moddef conv_branch_init(conv, branches):weight = conv.weightn = weight.size(0)k1 = weight.size(1)k2 = weight.size(2)nn.init.normal_(weight, 0, math.sqrt(2. / (n * k1 * k2 * branches)))nn.init.constant_(conv.bias, 0)def conv_init(conv):nn.init.kaiming_normal_(conv.weight, mode='fan_out')nn.init.constant_(conv.bias, 0)def bn_init(bn, scale):nn.init.constant_(bn.weight, scale)nn.init.constant_(bn.bias, 0)class unit_tcn(nn.Module):def __init__(self, in_channels, out_channels, kernel_size=9, stride=1):super(unit_tcn, self).__init__()pad = int((kernel_size - 1) / 2)self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=(kernel_size, 1), padding=(pad, 0),stride=(stride, 1))self.bn = nn.BatchNorm2d(out_channels)self.relu = nn.ReLU(inplace=True)conv_init(self.conv)bn_init(self.bn, 1)def forward(self, x):x = self.bn(self.conv(x))return xclass unit_gcn(nn.Module):def __init__(self, in_channels, out_channels, A, coff_embedding=4, num_subset=3, adaptive=True, attention=True):super(unit_gcn, self).__init__()inter_channels = out_channels // coff_embeddingself.inter_c = inter_channelsself.out_c = out_channelsself.in_c = in_channelsself.num_subset = num_subset#暫時不知道是什么？應(yīng)該是和后面的A有關(guān)，等我把第三部分寫完回來填坑……#回來填坑了，這個地方應(yīng)該指的是劃分子集的方式，按最初的ST-GCN的第三種方式，一共被劃分為3個子集，因此num_set=3#并且個st-gcn單元都有自己的權(quán)重參數(shù)用于訓練參見下圖

摘自https://www.cnblogs.com/shyern/p/11262926.html#_label4_1

num_jpts = A.shape[-1]self.conv_d = nn.ModuleList()#類似于pytho中的list類型，只是將一系列層裝入列表，并沒有實現(xiàn)forward()方法for i in range(self.num_subset):self.conv_d.append(nn.Conv2d(in_channels, out_channels, 1))#構(gòu)建3個卷積層if adaptive:#自適應(yīng)self.PA = nn.Parameter(torch.from_numpy(A.astype(np.float32)))#復(fù)制A，torch.from_numpy,The returned tensor and `ndarray` share the same memory. Modifications to the tensor will be reflected in the `ndarray` and vice versa. The returned tensor is not resizable.#后半部分nn.Parameter，將一個不可訓練的類型Tensor轉(zhuǎn)換成可以訓練的類型parameter并將這個parameter綁定到這個module里面self.alpha = nn.Parameter(torch.zeros(1))# self.beta = nn.Parameter(torch.ones(1))# nn.init.constant_(self.PA, 1e-6)# self.A = Variable(torch.from_numpy(A.astype(np.float32)), requires_grad=False)# self.A = self.PAself.conv_a = nn.ModuleList()self.conv_b = nn.ModuleList()for i in range(self.num_subset):self.conv_a.append(nn.Conv2d(in_channels, inter_channels, 1))self.conv_b.append(nn.Conv2d(in_channels, inter_channels, 1))#上面的部分一共構(gòu)建了3個卷積塊conv_d，conv_a，conv_b，每個卷積塊有3（num_subset）個卷積層else:self.A = Variable(torch.from_numpy(A.astype(np.float32)), requires_grad=False)self.adaptive = adaptiveif attention:#是否添加注意力機制# self.beta = nn.Parameter(torch.zeros(1))# self.gamma = nn.Parameter(torch.zeros(1))# unified attention# self.Attention = nn.Parameter(torch.ones(num_jpts))# temporal attention 時間維度注意力self.conv_ta = nn.Conv1d(out_channels, 1, 9, padding=4)#1d卷積nn.init.constant_(self.conv_ta.weight, 0)#初始化常量nn.init.constant_(self.conv_ta.bias, 0)# s attention #空間維度注意力ker_jpt = num_jpts - 1 if not num_jpts % 2 else num_jpts #ker_jpt始終為奇數(shù)pad = (ker_jpt - 1) // 2 #偶數(shù)填充self.conv_sa = nn.Conv1d(out_channels, 1, ker_jpt, padding=pad)nn.init.xavier_normal_(self.conv_sa.weight)nn.init.constant_(self.conv_sa.bias, 0)# channel attention 通道級注意力rr = 2self.fc1c = nn.Linear(out_channels, out_channels // rr)self.fc2c = nn.Linear(out_channels // rr, out_channels)nn.init.kaiming_normal_(self.fc1c.weight)#Xavier初始化方法nn.init.constant_(self.fc1c.bias, 0)nn.init.constant_(self.fc2c.weight, 0)nn.init.constant_(self.fc2c.bias, 0)# self.bn = nn.BatchNorm2d(out_channels)# bn_init(self.bn, 1)self.attention = attentionif in_channels != out_channels:#如果輸入通道與輸出通道不相同，則調(diào)整至out_channel大小self.down = nn.Sequential(nn.Conv2d(in_channels, out_channels, 1),nn.BatchNorm2d(out_channels))else:self.down = lambda x: xself.bn = nn.BatchNorm2d(out_channels)self.soft = nn.Softmax(-2)self.tan = nn.Tanh()self.sigmoid = nn.Sigmoid()self.relu = nn.ReLU(inplace=True)for m in self.modules():if isinstance(m, nn.Conv2d):conv_init(m)elif isinstance(m, nn.BatchNorm2d):bn_init(m, 1)bn_init(self.bn, 1e-6)for i in range(self.num_subset):conv_branch_init(self.conv_d[i], self.num_subset)

代碼里的N，V和圖里的正好反了，看的時候注意一下

#如何調(diào)用構(gòu)建的模型def forward(self, x):N, C, T, V = x.size()y = Noneif self.adaptive:A = self.PA# A = A + self.PAfor i in range(self.num_subset):A1 = self.conv_a[i](x).permute(0, 3, 1, 2).contiguous().view(N, V, self.inter_c * T)#上圖藍2A2 = self.conv_b[i](x).view(N, self.inter_c * T, V)#上圖藍3A1 = self.tan(torch.matmul(A1, A2) / A1.size(-1)) # N V V，上圖藍4A1 = A[i] + A1 * self.alphaA2 = x.view(N, C * T, V)#上圖藍1z = self.conv_d[i](torch.matmul(A2, A1).view(N, C, T, V))y = z + y if y is not None else z#上圖藍5else:A = self.A.cuda(x.get_device()) * self.maskfor i in range(self.num_subset):A1 = A[i]A2 = x.view(N, C * T, V)z = self.conv_d[i](torch.matmul(A2, A1).view(N, C, T, V))y = z + y if y is not None else zy = self.bn(y)y += self.down(x)y = self.relu(y) if self.attention:# spatial attentionse = y.mean(-2) # N C V，歸一化se1 = self.sigmoid(self.conv_sa(se))#計算注意力機制，下同不贅述y = y * se1.unsqueeze(-2) + y# a1 = se1.unsqueeze(-2)# temporal attentionse = y.mean(-1)se1 = self.sigmoid(self.conv_ta(se))y = y * se1.unsqueeze(-1) + y# a2 = se1.unsqueeze(-1)# channel attentionse = y.mean(-1).mean(-1)se1 = self.relu(self.fc1c(se))se2 = self.sigmoid(self.fc2c(se1))y = y * se2.unsqueeze(-1).unsqueeze(-1) + y# a3 = se2.unsqueeze(-1).unsqueeze(-1)# unified attention# y = y * self.Attention + y# y = y + y * ((a2 + a3) / 2)# y = self.bn(y)return y class TCN_GCN_unit(nn.Module):def __init__(self, in_channels, out_channels, A, stride=1, residual=True, adaptive=True, attention=True):super(TCN_GCN_unit, self).__init__()self.gcn1 = unit_gcn(in_channels, out_channels, A, adaptive=adaptive, attention=attention)#計算圖卷積self.tcn1 = unit_tcn(out_channels, out_channels, stride=stride)#計算時間卷積self.relu = nn.ReLU(inplace=True)# if attention:# self.alpha = nn.Parameter(torch.zeros(1))# self.beta = nn.Parameter(torch.ones(1))# temporal attention# self.conv_ta1 = nn.Conv1d(out_channels, out_channels//rt, 9, padding=4)# self.bn = nn.BatchNorm2d(out_channels)# bn_init(self.bn, 1)# self.conv_ta2 = nn.Conv1d(out_channels, 1, 9, padding=4)# nn.init.kaiming_normal_(self.conv_ta1.weight)# nn.init.constant_(self.conv_ta1.bias, 0)# nn.init.constant_(self.conv_ta2.weight, 0)# nn.init.constant_(self.conv_ta2.bias, 0)# rt = 4# self.inter_c = out_channels // rt# self.conv_ta1 = nn.Conv2d(out_channels, out_channels // rt, 1)# self.conv_ta2 = nn.Conv2d(out_channels, out_channels // rt, 1)# nn.init.constant_(self.conv_ta1.weight, 0)# nn.init.constant_(self.conv_ta1.bias, 0)# nn.init.constant_(self.conv_ta2.weight, 0)# nn.init.constant_(self.conv_ta2.bias, 0)# s attention# num_jpts = A.shape[-1]# ker_jpt = num_jpts - 1 if not num_jpts % 2 else num_jpts# pad = (ker_jpt - 1) // 2# self.conv_sa = nn.Conv1d(out_channels, 1, ker_jpt, padding=pad)# nn.init.constant_(self.conv_sa.weight, 0)# nn.init.constant_(self.conv_sa.bias, 0)# channel attention# rr = 16# self.fc1c = nn.Linear(out_channels, out_channels // rr)# self.fc2c = nn.Linear(out_channels // rr, out_channels)# nn.init.kaiming_normal_(self.fc1c.weight)# nn.init.constant_(self.fc1c.bias, 0)# nn.init.constant_(self.fc2c.weight, 0)# nn.init.constant_(self.fc2c.bias, 0)## self.softmax = nn.Softmax(-2)# self.sigmoid = nn.Sigmoid()self.attention = attentionif not residual:self.residual = lambda x: 0elif (in_channels == out_channels) and (stride == 1):self.residual = lambda x: xelse:self.residual = unit_tcn(in_channels, out_channels, kernel_size=1, stride=stride)def forward(self, x):if self.attention:y = self.relu(self.tcn1(self.gcn1(x)) + self.residual(x))# spatial attention# se = y.mean(-2) # N C V# se1 = self.sigmoid(self.conv_sa(se))# y = y * se1.unsqueeze(-2) + y# a1 = se1.unsqueeze(-2)# temporal attention# se = y.mean(-1) # N C T# # se1 = self.relu(self.bn(self.conv_ta1(se)))# se2 = self.sigmoid(self.conv_ta2(se))# # y = y * se1.unsqueeze(-1) + y# a2 = se2.unsqueeze(-1)# se = y # NCTV# N, C, T, V = y.shape# se1 = self.conv_ta1(se).permute(0, 2, 1, 3).contiguous().view(N, T, self.inter_c * V) # NTCV# se2 = self.conv_ta2(se).permute(0, 1, 3, 2).contiguous().view(N, self.inter_c * V, T) # NCVT# a2 = self.softmax(torch.matmul(se1, se2) / np.sqrt(se1.size(-1))) # N T T# y = torch.matmul(y.permute(0, 1, 3, 2).contiguous().view(N, C * V, T), a2) \# .view(N, C, V, T).permute(0, 1, 3, 2) * self.alpha + y# channel attention# se = y.mean(-1).mean(-1)# se1 = self.relu(self.fc1c(se))# se2 = self.sigmoid(self.fc2c(se1))# # y = y * se2.unsqueeze(-1).unsqueeze(-1) + y# a3 = se2.unsqueeze(-1).unsqueeze(-1)## y = y * ((a2 + a3) / 2) + y# y = self.bn(y)else:y = self.relu(self.tcn1(self.gcn1(x)) + self.residual(x))return y class Model(nn.Module):def __init__(self, num_class=60, num_point=25, num_person=2, graph=None, graph_args=dict(), in_channels=3,drop_out=0, adaptive=True, attention=True):super(Model, self).__init__()if graph is None:raise ValueError()else:Graph = import_class(graph)self.graph = Graph(**graph_args)A = self.graph.Aself.num_class = num_class#動作類別數(shù)self.data_bn = nn.BatchNorm1d(num_person * in_channels * num_point) #網(wǎng)絡(luò)模型構(gòu)建self.l1 = TCN_GCN_unit(3, 64, A, residual=False, adaptive=adaptive, attention=attention)self.l2 = TCN_GCN_unit(64, 64, A, adaptive=adaptive, attention=attention)self.l3 = TCN_GCN_unit(64, 64, A, adaptive=adaptive, attention=attention)self.l4 = TCN_GCN_unit(64, 64, A, adaptive=adaptive, attention=attention)self.l5 = TCN_GCN_unit(64, 128, A, stride=2, adaptive=adaptive, attention=attention)self.l6 = TCN_GCN_unit(128, 128, A, adaptive=adaptive, attention=attention)self.l7 = TCN_GCN_unit(128, 128, A, adaptive=adaptive, attention=attention)self.l8 = TCN_GCN_unit(128, 256, A, stride=2, adaptive=adaptive, attention=attention)self.l9 = TCN_GCN_unit(256, 256, A, adaptive=adaptive, attention=attention)self.l10 = TCN_GCN_unit(256, 256, A, adaptive=adaptive, attention=attention)self.fc = nn.Linear(256, num_class)nn.init.normal_(self.fc.weight, 0, math.sqrt(2. / num_class))bn_init(self.data_bn, 1)if drop_out:self.drop_out = nn.Dropout(drop_out)else:self.drop_out = lambda x: xdef forward(self, x):N, C, T, V, M = x.size()#對應(yīng)輸入數(shù)據(jù)格式，N=sample數(shù)量，C=（channel，3固定），T=幀，V=（關(guān)節(jié)數(shù)，固定25），M=（最大人數(shù)，固定2）x = x.permute(0, 4, 3, 1, 2).contiguous().view(N, M * V * C, T)#permute更換tensor維度的順序；#view通常會使用contiguous，與內(nèi)存有關(guān)，重新強制拷貝一份tensor#view順序取數(shù)組重新組成目標形狀x = self.data_bn(x)#self.data_bn = nn.BatchNorm1d(num_person * in_channels * num_point)x = x.view(N, M, V, C, T).permute(0, 1, 3, 4, 2).contiguous().view(N * M, C, T, V)#同上#model中定義的層結(jié)構(gòu)x = self.l1(x)x = self.l2(x)x = self.l3(x)x = self.l4(x)x = self.l5(x)x = self.l6(x)x = self.l7(x)x = self.l8(x)x = self.l9(x)x = self.l10(x)# N*M,C,T,Vc_new = x.size(1)x = x.view(N, M, c_new, -1)x = x.mean(3).mean(1)x = self.drop_out(x)#self.drop_out = nn.Dropout(drop_out)return self.fc(x)#全連接分類，self.fc = nn.Linear(256, num_class)

總結(jié)

以上是生活随笔為你收集整理的GCN代码超详解析Two-stream adaptive graph convolutional network for Skeleton-Based Action Recognition（二）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。