Unverified Commit 873db382 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

[Fix]: fix dcn bias bug in centerpoint (#137)

* fix dcn bias bug in centerpoint

* fix unit tesats bug of empty iou calculation
parent e830b38b
...@@ -10,7 +10,6 @@ model = dict( ...@@ -10,7 +10,6 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
...@@ -10,8 +10,7 @@ model = dict( ...@@ -10,8 +10,7 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
......
...@@ -10,7 +10,6 @@ model = dict( ...@@ -10,7 +10,6 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
...@@ -10,8 +10,7 @@ model = dict( ...@@ -10,8 +10,7 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
......
...@@ -10,7 +10,6 @@ model = dict( ...@@ -10,7 +10,6 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
...@@ -10,8 +10,7 @@ model = dict( ...@@ -10,8 +10,7 @@ model = dict(
out_channels=64, out_channels=64,
kernel_size=3, kernel_size=3,
padding=1, padding=1,
groups=4, groups=4),
bias=True),
init_bias=-2.19, init_bias=-2.19,
final_kernel=3))) final_kernel=3)))
......
...@@ -72,7 +72,7 @@ def test_max_iou_assigner_with_empty_gt(): ...@@ -72,7 +72,7 @@ def test_max_iou_assigner_with_empty_gt():
[5, 5, 15, 15], [5, 5, 15, 15],
[32, 32, 38, 42], [32, 32, 38, 42],
]) ])
gt_bboxes = torch.FloatTensor([]) gt_bboxes = torch.FloatTensor(size=(0, 4))
assign_result = self.assign(bboxes, gt_bboxes) assign_result = self.assign(bboxes, gt_bboxes)
expected_gt_inds = torch.LongTensor([0, 0, 0, 0]) expected_gt_inds = torch.LongTensor([0, 0, 0, 0])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment