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
Met this question when using predict.py. Error code location is located at
predict.py
i = order[0] keep.append(i)
after check the code, I found the key is to update squeeze() use
squeeze()
# before ids = (ovr<=threshold).nonzero().squeeze() # after ids = (ovr<=threshold).nonzero().squeeze(-1)
this change can solve the problem, when ids shape == (1,1), then squeeze(-1) return (1)
ids
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Met this question when using
predict.py
. Error code location is located atafter check the code, I found the key is to update
squeeze()
usethis change can solve the problem, when
ids
shape == (1,1), then squeeze(-1) return (1)The text was updated successfully, but these errors were encountered: