Deployment is in progres..

Project Information

Objective

To explore the application of data manipulation, transfer learning and built a custom deep neural network for a cutlery image dataset and then perform model evaluation to select the hero model to classify cutlery images into four categories - cup, knife, fork, spoon.

Achievements

  • Built a text dataset using web scraping and fine-tuned a pre-trained language model, achieving 85% test accuracy.

    • Tech Stack

    BeautifulSoup4, Tokenization, ALBERT, GPT-J, GPT-Sw3