waldeland / CNN-for-ASI

Tutorial: Convolutional Neural Networks for Automated Seismic Interpretation
Other
149 stars 79 forks source link

Doubt regarding usage #11

Closed akshitac8 closed 5 years ago

akshitac8 commented 5 years ago

Hello Dr Waldeland,

I have been trying to work through your code on Penobscot interpretation dataset for facies classification. I had a few questions:

  1. I need to work with multi-label classification. So, I need to store a single image in train folder with all the classes drawn in that image or I need to store separate image files. Thank you in advance :slightly_smiling_face:
waldeland commented 5 years ago

Hi, that should be no problem. Just use different colors for each class and update the color_table in data.py accordingly.

waldeland commented 5 years ago

https://github.com/waldeland/CNN-for-ASI/blob/b497ce17c1ee0f0499eba459eba85e263ca522c3/data.py#L131