def get_adjacency_matrix(self, labeling_mode=None):
if labeling_mode is None:
return self.A
if labeling_mode == 'spatial':
A = tools.get_spatial_graph(num_node, self_link, inward, outward)
elif labeling_mode == 'GIN':
A = tools.get_spatial_graph(num_node,
self_link,
inward,
outward,
normalize=False)
A = A[1:]
else:
raise ValueError()
return A
def get_adjacency_matrix(self, labeling_mode=None):
if labeling_mode is None:
return self.A
elif labeling_mode == 'spatial':
A = tools.get_spatial_graph(num_node, self_link, inward, outward)
elif labeling_mode == 'parts':
A = tools.get_part_based_graph(num_node,
self_link,
parts=[head, hands, torso, legs])
else:
raise ValueError()
return A
def get_adjacency_matrix(self, labeling_mode=None):
if labeling_mode is None:
return self.A
if labeling_mode == 'spatial':
A = []
for i in range(len(self.num_node)):
A1 = tools.get_spatial_graph(self.num_node[i],
self.self_link[i], self.inward[i],
self.outward[i])
A.append(A1)
else:
raise ValueError()
return A
def get_adjacency_matrix(self, labeling_mode=None):
if labeling_mode is None:
return self.A
if labeling_mode == 'spatial':
A = tools.get_spatial_graph(num_node, self_link, inward, outward)
elif labeling_mode == 'spatial_pre':
A = tools.get_spatial_graph_pre(num_node, self_link, inward,
outward, self.drop_list)
elif labeling_mode == 'spatial_post':
A = tools.get_spatial_graph_post(num_node, self_link, inward,
outward, self.drop_list)
else:
raise ValueError()
return A
def get_adjacency_matrix(self, labeling_mode=None):
if labeling_mode is None:
return self.A
if labeling_mode == 'spatial':
A = tools.get_spatial_graph(num_node, self_link, inward, outward)
else:
raise ValueError()
A[0, 3, 3] += 0.2
A[0, 4, 4] += 0.2
A[0, 6, 6] += 0.2
A[0, 7, 7] += 0.2
A[0, 9, 9] += 0.2
A[0, 10, 10] += 0.2
A[0, 12, 12] += 0.2
A[0, 13, 13] += 0.2
return A # A是三个图，依次是自身图、内图、外图。A：3*num_node*num_node
def get_A9(self):
A = tools.get_spatial_graph(9, self_link9, inward9, outward9)
return A
def get_A15(self):
A = tools.get_spatial_graph(13, self_link13, inward13, outward13)
return A
