Giter VIP home page Giter VIP logo

transfer-learning's Introduction

Transfer-Learning

Inception V3 for Transfer Learning on Cats and Dogs

File to train the Inception v3 model on any dataset is added(transfer.py)

I have added the weights file and the predict file(predict.py)

Instructions to use the predict file:

Recommended to use Anaconda 3

Dependencies:

  1. Tensorflow = 1.2.1
  2. Keras = 2+

Installing Tensorflow-cpu : pip install tensorflow

Installing keras: pip install keras

Project file structure

Transfer-Learning
| inception.model 
| README.md
| predict.py
| transfer.py
|__test_set
|   |__cats
|   |    cat01.jpg
|   |    cat02.jpg
|   |    ...
|   |__dogs
|   |    dog01.jpg
|   |    dog02.jpg
|   |    ..
|
|
|__training_set
|   |__cats
|   |    cat01.jpg
|   |    cat02.jpg
|   |    ...
|   |__dogs
|   |    dog01.jpg
|   |    dog02.jpg
|   |    ..

Operations with transfer.py

  1. Add the files in training_set and test_set

  2. Run python transfer.py --nb_epoch 5 --batch_size 320 --plot --output_model_file filename.model

  3. Later use the saved weights to predict any Image of cat or dog from the Internet or saved Images.

Operations for predict.py

  1. For Images saved locally python predict.py --image imagename.jpg --model_name inception.model

  2. For Images on the net python predict.py --image_url www.imagename.jpg --model_name inception.model

transfer-learning's People

Contributors

dhavalthkkar 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.