darshan-analytics / multilabel_image_classification_using_CNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

multilabel_image_classification_using_CNN

  1. This project is to classify MultiLabel data using CNN.
  2. This is made for used with transfer learning.
  3. The CNN will classify the data as Bear, Bearcats or Cats.
  4. Used efficientnetB0 along with multi-stage unfreezing to tune the CNN.
  5. Got accuracy around 96% for the validation set.

About


Languages

Language:Jupyter Notebook 100.0%