When I ran 'test.py', I encountered an error which read
# RuntimeError: boxes should have 4 elements in dimension 1, got 5.
The problem was caused by the following line in test_utils.py, line 46:
keep = nms(torch.tensor(dets_j), torch.tensor(scores_j), torch.tensor(cfg.TEST.NMS)).
I believe the issue may be resolved by changing the line to:
keep = nms(torch.tensor(boxes_j), torch.tensor(scores_j), torch.tensor(cfg.TEST.NMS))
Although I'm aware that I should submit a pull request, I have already made several changes to the code.
Thank you very much for your helpful code.
When I ran 'test.py', I encountered an error which read
# RuntimeError: boxes should have 4 elements in dimension 1, got 5.The problem was caused by the following line in test_utils.py, line 46:
keep = nms(torch.tensor(dets_j), torch.tensor(scores_j), torch.tensor(cfg.TEST.NMS)).I believe the issue may be resolved by changing the line to:
keep = nms(torch.tensor(boxes_j), torch.tensor(scores_j), torch.tensor(cfg.TEST.NMS))Although I'm aware that I should submit a pull request, I have already made several changes to the code.
Thank you very much for your helpful code.