microsoft / seismic-deeplearning

Deep Learning for Seismic Imaging and Interpretation
MIT License
403 stars 142 forks source link

debug multiple classes showing up on binary synthetic data - facilitates correct mask visualizations representative of the number of classes #326

Closed yalaudah closed 4 years ago

yalaudah commented 4 years ago

Investigate why the model predictions in the binary dataset result in images with multiple classes like this image: image image

This might be due to the logits being passed on to mask_to_disk() before taking the max()

yalaudah commented 4 years ago

I removed the other issue (binary image becomes non binary after augmentation) since it is unrelated. It is now referenced in issue: #344