We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Thank's for your work, I try hnms in my object detection predictions. `import torch from hnms import MultiHNMS hnms = MultiHNMS(num=1, alpha=0.7)
hnms
xywh = [[10, 20, 10, 20], [10, 20, 10, 20], [30, 6, 4, 5]] conf = [0.9, 0.8, 0.9] xywh = torch.tensor(xywh).float() conf = torch.tensor(conf) keep = hnms(xywh, conf) print(keep)`
why the output keep is one tensor number, I think it should be some indexes of keeped boxes.
keep
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Thank's for your work, I try
hnms
in my object detection predictions.`import torch
from hnms import MultiHNMS
hnms = MultiHNMS(num=1, alpha=0.7)
center x, center y, width, height
xywh = [[10, 20, 10, 20], [10, 20, 10, 20], [30, 6, 4, 5]]
conf = [0.9, 0.8, 0.9]
xywh = torch.tensor(xywh).float()
conf = torch.tensor(conf)
keep = hnms(xywh, conf)
print(keep)`
why the output
keep
is one tensor number, I think it should be some indexes of keeped boxes.The text was updated successfully, but these errors were encountered: