revert anchor generator

wonjuleee · wonjuleee · commit 5cf5f747fba9 · 2024-02-08T15:07:16.000+09:00
diff --git a/src/datumaro/plugins/anchor_generator.py b/src/datumaro/plugins/anchor_generator.py
@@ -183,7 +183,7 @@ def get_anchors(
                 x_center + 0.5 * ws,
                 y_center + 0.5 * hs,
             ]
-            base_anchors = torch.stack(base_anchors, dim=-1).to(self.device)
+            base_anchors = torch.stack(base_anchors, dim=-1)
 
             anchors = base_anchors[None, :, :] + shifts[:, None, :]
             anchors = anchors.view(-1, 4)

Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ def get_anchors(`
`183`	`183`	`x_center + 0.5 * ws,`
`184`	`184`	`y_center + 0.5 * hs,`
`185`	`185`	`]`
`186`		`- base_anchors = torch.stack(base_anchors, dim=-1).to(self.device)`
	`186`	`+ base_anchors = torch.stack(base_anchors, dim=-1)`
`187`	`187`
`188`	`188`	`anchors = base_anchors[None, :, :] + shifts[:, None, :]`
`189`	`189`	`anchors = anchors.view(-1, 4)`