-
Notifications
You must be signed in to change notification settings - Fork 3.9k
MedShapeNet and example #9823
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
MedShapeNet and example #9823
Conversation
|
“MedShapeNet is not divided by train and test by default. Therefore, train data is downloaded from the first n required number of samples, and the test data is downloaded from the last n required number of samples.” i would recommend the data be split randomly if train/val/test is not predefined |
note that CI is also failing with this issue. however, I am having trouble installing medshapenet through pip to verify if your code actually works. Is that how you install it? could you provide your setup commands? based on this i will also advise wether those instructions should be included in your code somewhere (i will decide based on the complexity of the set up instructions) |
0825c61 to
e2f7faf
Compare
Codecov Report✅ All modified and coverable lines are covered by tests. Additional details and impacted files@@ Coverage Diff @@
## master #9823 +/- ##
==========================================
- Coverage 86.11% 85.43% -0.68%
==========================================
Files 496 496
Lines 33655 34007 +352
==========================================
+ Hits 28981 29055 +74
- Misses 4674 4952 +278 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
237bd37 to
e2f7faf
Compare
a20df74 to
8a1e2d4
Compare
85fda59 to
1da4e60
Compare
9a25a82 to
028ff15
Compare
7feffdb to
73e96d3
Compare
75d9ce2 to
033987e
Compare
|
Should it be |
|
Indeed, The problem is that MedShapeNet uses a down-grade version of numpy. By default, it downloads version I'm not sure how to proceed with this conflict. Any hint you can give will be welcome. |
|
@xnuohz and @puririshi98 I have talked with the MedShapeNet. Looks like they already solved the problem, I was able to run the Dataset on Python 3.10.12. Please corroborate and let me know if there is some pending issue to solve |
xnuohz
left a comment
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.
left some comments:)
|
Thanks for your time checking the code and adding the last suggestions. I have already worked in the suggested changes:
|
|
@jdhenaos thank you. can you share the train log? |
|
@xnuohz the train log: |
xnuohz
left a comment
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.
add an unit test if the dataset is small and easy to download
puririshi98
left a comment
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 will merge once CI passes
I have added MedShapeNet dataset, adding eight classes. The next folders were ignored from MedShapeNet Dataset due to a lack of samples: ASOCA (n=43) AVT (n=45) AutoImplantCraniotomy (n=14) FaceVR (n=14) This Dataset requires a size number as a parameter to download the same number of samples per shape to ensure class balance. MedShapeNet is not divided by train and test by default. Therefore, train data is downloaded from the first n required number of samples, and the test data is downloaded from the last n required number of samples. dgcnn classification example was modified to use both ModelNet and MedShapeNet. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Rishi Puri <[email protected]> Co-authored-by: Rishi Puri <[email protected]>
|
|
||
| subset = [] | ||
| for dataset in list_of_datasets: | ||
| self.newpath = self.root + '/' + dataset.split("/")[1] |
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.
I got to this PR while reviewing #10472. Does this work on Windows? Could we make it os-agnostic?
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.
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.
I have added MedShapeNet dataset, adding eight classes.
The next folders were ignored from MedShapeNet Dataset due to a lack of samples:
ASOCA (n=43)
AVT (n=45)
AutoImplantCraniotomy (n=14)
FaceVR (n=14)
This Dataset requires a size number as a parameter to download the same number of samples per shape to ensure class balance.
MedShapeNet is not divided by train and test by default. Therefore, train data is downloaded from the first n required number of samples, and the test data is downloaded from the last n required number of samples.
dgcnn classification example was modified to use both ModelNet and MedShapeNet.