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