llSourcell / AI_Startup_Prototype

This is the code for "Watch Me Build an AI Startup" By Siraj Raval on Youtube
368 stars 264 forks source link

Trained Model is Classification But result shown is more of segmentation. #9

Open harveenchadha opened 5 years ago

harveenchadha commented 5 years ago

Hi,

Thank you for your efforts, but I want to point out that your model output is classification i.e. 3 outcomes but the image you showed is more of a segmentation task that requires something like mask-rcnn sort of thing and similarly the data according to that which can be found here:

https://www.kaggle.com/c/rsna-pneumonia-detection-challenge/data