|
17 | 17 | boxes = torch.tensor([[0, 0, 20, 20], [0, 0, 0, 0],
|
18 | 18 | [10, 15, 30, 35], [23, 35, 93, 95]], dtype=torch.float)
|
19 | 19 |
|
20 |
| -masks = torch.tensor([ |
21 |
| - [ |
22 |
| - [-2.2799, -2.2799, -2.2799, -2.2799, -2.2799], |
23 |
| - [5.0914, 5.0914, 5.0914, 5.0914, 5.0914], |
24 |
| - [-2.2799, -2.2799, -2.2799, -2.2799, -2.2799], |
25 |
| - [-2.2799, -2.2799, -2.2799, -2.2799, -2.2799], |
26 |
| - [-2.2799, -2.2799, -2.2799, -2.2799, -2.2799] |
27 |
| - ], |
28 |
| - [ |
29 |
| - [5.0914, 5.0914, 5.0914, 5.0914, 5.0914], |
30 |
| - [-2.2799, -2.2799, -2.2799, -2.2799, -2.2799], |
31 |
| - [5.0914, 5.0914, 5.0914, 5.0914, 5.0914], |
32 |
| - [5.0914, 5.0914, 5.0914, 5.0914, 5.0914], |
33 |
| - [-1.4541, -1.4541, -1.4541, -1.4541, -1.4541] |
34 |
| - ], |
35 |
| - [ |
36 |
| - [-1.4541, -1.4541, -1.4541, -1.4541, -1.4541], |
37 |
| - [-1.4541, -1.4541, -1.4541, -1.4541, -1.4541], |
38 |
| - [-1.4541, -1.4541, -1.4541, -1.4541, -1.4541], |
39 |
| - [-1.4541, -1.4541, -1.4541, -1.4541, -1.4541], |
40 |
| - [5.0914, 5.0914, 5.0914, 5.0914, 5.0914], |
41 |
| - ] |
42 |
| -], dtype=torch.float) |
43 |
| - |
44 | 20 |
|
45 | 21 | class Tester(unittest.TestCase):
|
46 | 22 |
|
|
0 commit comments