7 lines
316 B
Python
7 lines
316 B
Python
|
import torch
|
||
|
|
|
||
|
|
|
||
|
|
def pose_edge_index():
|
||
|
|
start = [15, 14, 13, 12, 19, 18, 17, 16, 0, 1, 2, 3, 8, 9, 10, 3, 4, 5, 6, 8, 8, 4, 20, 21, 21, 22, 24, 22]
|
||
|
|
end = [14, 13, 12, 0, 18, 17, 16, 0, 1, 2, 3, 8, 9, 10, 11, 4, 5, 6, 7, 4, 20, 20, 21, 22, 24, 23, 25, 24]
|
||
|
|
return torch.tensor([start+end, end+start])
|