Spaces:
Running
Running
File size: 610 Bytes
6f6830f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# Copyright (c) Facebook, Inc. and its affiliates.
import unittest
import torch
from detectron2.structures.keypoints import Keypoints
class TestKeypoints(unittest.TestCase):
def test_cat_keypoints(self):
keypoints1 = Keypoints(torch.rand(2, 21, 3))
keypoints2 = Keypoints(torch.rand(4, 21, 3))
cat_keypoints = keypoints1.cat([keypoints1, keypoints2])
self.assertTrue(torch.all(cat_keypoints.tensor[:2] == keypoints1.tensor).item())
self.assertTrue(torch.all(cat_keypoints.tensor[2:] == keypoints2.tensor).item())
if __name__ == "__main__":
unittest.main()
|