|
|
|
import unittest |
|
import torch |
|
|
|
from detectron2.structures import Instances |
|
|
|
|
|
class TestInstancesIndexing(unittest.TestCase): |
|
def test_int_indexing(self): |
|
attr1 = torch.tensor([[0.0, 0.0, 1.0], [0.0, 0.0, 0.5], [0.0, 0.0, 1.0], [0.0, 0.5, 0.5]]) |
|
attr2 = torch.tensor([0.1, 0.2, 0.3, 0.4]) |
|
instances = Instances((100, 100)) |
|
instances.attr1 = attr1 |
|
instances.attr2 = attr2 |
|
for i in range(-len(instances), len(instances)): |
|
inst = instances[i] |
|
self.assertEqual((inst.attr1 == attr1[i]).all(), True) |
|
self.assertEqual((inst.attr2 == attr2[i]).all(), True) |
|
|
|
self.assertRaises(IndexError, lambda: instances[len(instances)]) |
|
self.assertRaises(IndexError, lambda: instances[-len(instances) - 1]) |
|
|
|
|
|
if __name__ == "__main__": |
|
unittest.main() |
|
|
