/CNN_From_Scratch

Binary classifier CNN model from scratch using PyTorch, classifies Bananas and Pineapples

Primary LanguageJupyter NotebookMIT LicenseMIT

Classifies Images into 2 classes, Banana and Pineapple

Uses opencv instead of builtin torch functions,trained on CPU, data scraped from Google and Yandex (included in repo)