def get_edge(self, layout):
    if (layout == 'openpose'):
        self.num_node = 18
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_link = [(4, 3), (3, 2), (7, 6), (6, 5), (13, 12), (12, 11), (10, 9), (9, 8), (11, 5), (8, 2), (5, 1), (2, 1), (0, 1), (15, 0), (14, 0), (17, 15), (16, 14)]
        self.edge = (self_link + neighbor_link)
        self.center = 1
    elif (layout == 'ntu-rgb+d'):
        self.num_node = 25
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_1base = [(1, 2), (2, 21), (3, 21), (4, 3), (5, 21), (6, 5), (7, 6), (8, 7), (9, 21), (10, 9), (11, 10), (12, 11), (13, 1), (14, 13), (15, 14), (16, 15), (17, 1), (18, 17), (19, 18), (20, 19), (22, 23), (23, 8), (24, 25), (25, 12)]
        neighbor_link = [((i - 1), (j - 1)) for (i, j) in neighbor_1base]
        self.edge = (self_link + neighbor_link)
        self.center = (21 - 1)
    elif (layout == 'ntu_edge'):
        self.num_node = 24
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_1base = [(1, 2), (3, 2), (4, 3), (5, 2), (6, 5), (7, 6), (8, 7), (9, 2), (10, 9), (11, 10), (12, 11), (13, 1), (14, 13), (15, 14), (16, 15), (17, 1), (18, 17), (19, 18), (20, 19), (21, 22), (22, 8), (23, 24), (24, 12)]
        neighbor_link = [((i - 1), (j - 1)) for (i, j) in neighbor_1base]
        self.edge = (self_link + neighbor_link)
        self.center = 2
    else:
        raise ValueError('Do Not Exist This Layout.')
def get_edge(self, layout):
    if (layout == 'openpose'):
        self.num_node = 18
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_link = [(4, 3), (3, 2), (7, 6), (6, 5), (13, 12), (12, 11), (10, 9), (9, 8), (11, 5), (8, 2), (5, 1), (2, 1), (0, 1), (15, 0), (14, 0), (17, 15), (16, 14)]
        self.edge = (self_link + neighbor_link)
        self.center = 1
    elif (layout == 'ntu-rgb+d'):
        self.num_node = 25
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_1base = [(1, 2), (2, 21), (3, 21), (4, 3), (5, 21), (6, 5), (7, 6), (8, 7), (9, 21), (10, 9), (11, 10), (12, 11), (13, 1), (14, 13), (15, 14), (16, 15), (17, 1), (18, 17), (19, 18), (20, 19), (22, 23), (23, 8), (24, 25), (25, 12)]
        neighbor_link = [((i - 1), (j - 1)) for (i, j) in neighbor_1base]
        self.edge = (self_link + neighbor_link)
        self.center = (21 - 1)
    elif (layout == 'ntu_edge'):
        self.num_node = 24
        self_link = [(i, i) for i in range(self.num_node)]
        neighbor_1base = [(1, 2), (3, 2), (4, 3), (5, 2), (6, 5), (7, 6), (8, 7), (9, 2), (10, 9), (11, 10), (12, 11), (13, 1), (14, 13), (15, 14), (16, 15), (17, 1), (18, 17), (19, 18), (20, 19), (21, 22), (22, 8), (23, 24), (24, 12)]
        neighbor_link = [((i - 1), (j - 1)) for (i, j) in neighbor_1base]
        self.edge = (self_link + neighbor_link)
        self.center = 2
    else:
        raise ValueError('Do Not Exist This Layout.')