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
Predict on DDSM #13
Comments
@Chunlwu, thanks for your kind words. First of all, did you notice that we updated the code a few weeks ago? There was a small bug in presenting the output. Please make sure that you use the code after that fix. Regarding compatibility with DDSM, we didn't do such an experiment. I would be very curious to see your results when you are finished. Assuming that you did all of the preprocessing the same way as we did and you are using the code after the fix, there a few things to consider.
I think probably the best way to use our network on DDSM would be to fine-tune it using DDSM to make sure that the network is adjusted to the shift in the distribution and to the shift in the definition of the labels. |
@kjgeras Thanks for you kindly reply~ I will finetune model on DDSM, and try again~ |
@WajeehaAnsar You should convert LJPEG to PNG, and then add new records into exam_list_before_cropping.pkl. that's all for predicting. |
@Chunlwu do you finetune model on DDSM? what is the result of your model |
@Chunlwu , hey did you finetune your model for DDSM? |
Firstly, Thanks for your great work on mammogram classification. Recently, I tried to predict your model on a public dataset(DDSM, and CBIS-DDSM). But I found the result is always predicted as BENIGN. Below is a sample case for your reference
Predicted by model (only image):
left_benign right_benign left_malignant right_malignant
0.2456 0.3804 0.0131 0.0716
0.1495 0.5369 0.0180 0.1072
0.1644 0.1658 0.0338 0.0284
0.1821 0.3101 0.0121 0.0585
GoundTurth:
left_benign right_benign left_malignant right_malignant
1 1 0 0
0 0 1 1
0 0 1 1
0 0 0 0
I known the imbalance issue which described in #9, so I selected 2 obvious MALIGNANT cases and 1 obvious BENIGN case. And done all preprocessing which described in #9 and dataset report. But the result is still predicted as BENIGN.
So, could you have evaluated the released model (in your code: breast_cancer_classifier/models/sample_image_model.p) on DDSM or INBreast? And the other question I want to known is, what's different between DDSM with your private dataset?
Thanks
The text was updated successfully, but these errors were encountered: