We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 36ccfb1 commit 1b4ef0aCopy full SHA for 1b4ef0a
tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py
@@ -171,10 +171,10 @@ def test_forward_train(self):
171
@e2e_pytest_unit
172
def test_simple_test_bboxes(self):
173
feats = [
174
- torch.randn([2, 256, 100, 134]),
175
- torch.randn([2, 256, 50, 67]),
176
- torch.randn([2, 256, 25, 34]),
177
- torch.randn([2, 256, 13, 17]),
+ torch.zeros([2, 256, 100, 134]),
+ torch.zeros([2, 256, 50, 67]),
+ torch.zeros([2, 256, 25, 34]),
+ torch.zeros([2, 256, 13, 17]),
178
]
179
img_metas = [
180
{
@@ -208,6 +208,7 @@ def test_simple_test_bboxes(self):
208
209
self.bbox_head.eval()
210
results = self.bbox_head.simple_test_bboxes(feats, img_metas)
211
+ breakpoint()
212
assert len(results) == 2
213
assert results[0][0].shape == torch.Size([300, 5])
214
assert results[0][1].shape == torch.Size([300])
0 commit comments