Try it out on Kaggle: https://www.kaggle.com/code/aadipatangi/freyhacks-image-setup
We were inspired to build an ice-cream classifier by the theme of summer and what we appreciate most about it, a good ol' scoop of ice cream.
Using Tensorflow, we built a model that can classify between 4 different brands of ice cream: Häagen-Dazs, Breyers, Ben & Jerry's, and Talenti.
We built our model by creating a convolutional neural network with 3 layers - 1 Flatten and 2 dense layers. We trained it based on a dataset of ice cream images from Kaggle [https://www.kaggle.com/datasets/tysonpo/ice-cream-dataset]
Some challenges that we ran into were installing packages. For a reason that we are still unclear about, Tensorflow was not installing on our Visual Studio Code which is why we switched to Kaggle. Kaggle was sluggish and sometimes it was frustrating to wait for it to catch up.
We are proud that we created a functional neural network. We are both newbies in the field of machine learning.
We learned how to build a neural network. We also learned how to properly do time management. I personally learned how to make a youtube video.
We want to make it available on the web possibly using flask.