-
Notifications
You must be signed in to change notification settings - Fork 1.6k
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
[Feature] Add S3DIS dataset for 3D object detection #835
Conversation
Codecov Report
@@ Coverage Diff @@
## master #835 +/- ##
==========================================
+ Coverage 49.18% 49.23% +0.05%
==========================================
Files 210 210
Lines 15986 16019 +33
Branches 2552 2556 +4
==========================================
+ Hits 7862 7887 +25
- Misses 7625 7630 +5
- Partials 499 502 +3
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
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.
Good job! @filaPro two questions:
- Similar to ScanNet detection, S3DIS detection also doesn't involve
yaw
prediction right? So the boxes here are all axis-aligned. - Maybe we can move the two S3DIS detection datasets code to be above the two S3DIS segmentation datasets code? This follows
scannet_dataset.py
The browse_dataset
result is great! Only need to add some docstring and minor code structure refinement. Also unit tests are needed.
I let @THU17cyz have a look here because this may involve our progress of refactoring coordinate system. |
I also removed _S3DISDataset in favour of ConcatDataset. This also requires a fmall fix in |
Great work, many thanks to @filaPro ! Similar to ScanNet det, there is no conflict with our ongoing coordinate system refactor. |
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.
Almost ready to merge. Only need to add some unit tests.
Don't quite understand lint problems, so revert |
This is because sometimes |
Ready to merge after unit tests are added |
Also updated test data file for S3DIS. |
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.
Good Job @filaPro! Cheers!
Close #817.
@Wuziyi616 can you have a look about general structure? Then I can also add unit test and documentation. I'm also attaching visualization from
browse_dataset
.