-
Notifications
You must be signed in to change notification settings - Fork 212
add object detection prediction example and fix batch #283
Conversation
4433656
to
cab4952
Compare
cab4952
to
9ebe92e
Compare
Codecov Report
@@ Coverage Diff @@
## master #283 +/- ##
==========================================
+ Coverage 82.48% 88.22% +5.73%
==========================================
Files 74 74
Lines 3575 3576 +1
==========================================
+ Hits 2949 3155 +206
+ Misses 626 421 -205
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
534281a
to
4b877c8
Compare
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.
LGTM !
4b877c8
to
95aed15
Compare
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.
LGTM
@@ -243,6 +243,8 @@ def default_uncollate(batch: Any): | |||
batch_type = type(batch) | |||
|
|||
if isinstance(batch, Tensor): | |||
if len(batch.shape) == 0: # 0 shape tensors | |||
return batch |
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.
Any reason not to return a list here? (like [batch]
)
What does this PR do?
Add object detection example #277
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃