Skip to content

Commit 20a6625

Browse files
Leon Dongfacebook-github-bot
Leon Dong
authored andcommitted
Fix detector build issues
Summary: Made changes to make some previous components compatible with vizard. - Modified rotated box to accept additional arguments to support forward compatibility with vizard (arguments are unused) - Created new config file for testing, don't need to change and revert changes golden config file anymore during devloop - Added registering my test train datasets to setup function - Added imports to load components into registries - **[IMPORTANT TO REVIEW] Modified serializer class for datasets (no longer uses SharedList, read more on what SharedList does here: https://fburl.com/code/23u0ci1z)** Reviewed By: srihari2761 Differential Revision: D64119684 fbshipit-source-id: 662cf1a271cd5db6686c03d6ed93dd85e5287f9a
1 parent ebe8b45 commit 20a6625

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

detectron2/structures/rotated_boxes.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Copyright (c) Facebook, Inc. and its affiliates.
22
import math
3-
from typing import List, Tuple
3+
from typing import Any, List, Tuple
44
import torch
55

66
from detectron2.layers.rotated_boxes import pairwise_iou_rotated
@@ -229,7 +229,7 @@ def clone(self) -> "RotatedBoxes":
229229
"""
230230
return RotatedBoxes(self.tensor.clone())
231231

232-
def to(self, device: torch.device):
232+
def to(self, device: torch.device, *args: Any, **kwargs: Any):
233233
# Boxes are assumed float32 and does not support to(dtype)
234234
return RotatedBoxes(self.tensor.to(device=device))
235235

0 commit comments

Comments
 (0)