Finally, we try it on me. Again, correct prediction but not very confident.
We can see there is some limitation to the model we have created. There are two major barriers:
- Imperfect data to train the model.
- We are also using a CNN from scratch.
The imperfect data is self-explanatory. The issue is our model is fresh. It has to learn what edges, colours, patterns from scratch. We can use transfer learning. This is where a pre-trained model can be used instead. A pre-trained model is already astute to the basics of what is in an image and so data is better utilised for our objective: eye colour.
I hope you found this interesting. Please let me know if you would like your photo tested against this model. Next, we will discuss the ethics of technology.
Thanks for reading and all the best for the week ahead.
Ngā mihi nui,