Code Monkey home page Code Monkey logo

image_classification's Introduction

Image_Classification

Build an Image classification model for flower species

Objective of this notebook

  • The purpose of this notebook is to build a Image classifier which would be capable of predicting the species of flowers
  • We compare and contrast the performance of various machine learning algorithms VS Deep learning algorithms in an effort to demostrate how traditional ML algorithms perform poorly on Image Data
  • Details of the problem statement , data set , summary of the code/solution , sample output/Prediction from the program and final result of the project are listed in the sections to follow.

Problem Statement

Computer Vision can be used to classify images .In this partcular use case , a university conducting research involving the understanding of the characteristics of flowers,require an automation which can create a classifier capable of determining a flower’s species from a photo

Data Description:

The dataset comprises of images from 17 plant species. It can be downloaded from TensorFlow [tflearn.datasets.oxflower17 as oxflower17 ]

Domain:

Botanical research

Summary of the Solution/Code:

The code aims at building a image classfifier

  • We begin by doing an Exploratory Data analyses and Visualisation/viewing of the images
  • We then do the required pre-processing for the data to make it compatible with the various models to be built.Each model would require the data to be prepared specifically for that model to ensure compatibility.
  • Essentially , the approach used in this worksheet is to try out multiple algorithms to build the classifier and choose the best contender
  • Multiple 'traditional' ML algorithms like LR,KNN,SVM,Random Forest,Ada Boost, Gradient Boostare tried and results were recorded
  • We then try deep learning algorithms like ANN and a CNN(build custom ANN & CNN networks) and try to solve the problem at hand .Results are recorded
  • Finally we try two pretrained CNN with a base of VGG16 and RESNET50 respectively and observe that CNN with base VGG16 gives us the best results
  • Pre-trained CNN(on ImageNet) with VGG16 Network as a base performed the best giving us 87% on Test Data
  • Refer python worksheet Project_P4_ImageClassfication_BotanicalResearch_Flowers.ipynb for the solution

Sample Ouput/Prediction :

Here is a sample result/output from the program/model

image image

Result

image

image

References & Guidance

  • PGP Course Material
  • Evaluator Feedback

image_classification's People

Contributors

radhika-keni avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.