-
Notifications
You must be signed in to change notification settings - Fork 1.3k
New issue
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
fix assertion #142
fix assertion #142
Conversation
Codecov Report
@@ Coverage Diff @@
## master #142 +/- ##
=======================================
Coverage 81.67% 81.67%
=======================================
Files 86 86
Lines 5026 5026
Branches 870 870
=======================================
Hits 4105 4105
Misses 767 767
Partials 154 154
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
@@ -105,7 +106,7 @@ def transform_preds(coords, center, scale, output_size): | |||
Returns: | |||
np.ndarray: Predicted coordinates in the images. | |||
""" | |||
assert coords.shape[1] == 2 or coords.shape[1] == 5 | |||
assert coords.shape[1] == 2 or coords.shape[1] == 4 or coords.shape[1] == 5 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
coords.shape[1] in (2, 4, 5)
Co-authored-by: lizz <[email protected]>
Co-authored-by: lizz <[email protected]>
#138 fix assertion error.