ScanLori - Machine Learning

This repository have both files and dataset for Image Classsification and Object Detection.

Datasets

For scalability purpose, the dataset consists of 37 classes for Image Classification and 8 for Object Detection.

Object detection used the help of RoboFlow API to annotate therefore there is no need for uploading the Dataset. The dataset is saved under the Roboflow(api_key) line

Models

Image Classification model was built from ground up using CNN model with resnet50 and Batch Normalization

Object Detection model was built using transfer learning with YoLoV8