Giter VIP home page Giter VIP logo

liyaunknown / mobilenet_v2 Goto Github PK

View Code? Open in Web Editor NEW
2.0 2.0 0.0 10.18 MB

Implementation of Mobilenet V2 for binary image classification of dogs and cats using Keras and TensorFlow. ๐Ÿ“š Trained on a dataset of dogs and cats images, with customizable scripts for training, testing, and prediction on new data. ๐Ÿ“Š๐Ÿ› ๏ธ

Home Page: https://medium.com/@liyafarahani

Python 0.14% Jupyter Notebook 99.86%
data-science dataset mobilenet-v2 python tensorflow

mobilenet_v2's Introduction

MobileNet_V2

Implementation of Mobilenet V2 for binary image classification of dogs and cats using Keras and TensorFlow. ๐Ÿ“š Trained on a dataset of dogs and cats images, with customizable scripts for training, testing, and prediction on new data. ๐Ÿ“Š๐Ÿ› ๏ธ

Dogs and Cats Classification with Mobilenet V2

Summary: This code implements a binary image classification model for dogs ๐Ÿถ and cats ๐Ÿฑ using the Mobilenet V2 architecture in TensorFlow and Keras. The "cats_vs_dogs" dataset from TensorFlow Datasets is used to train, validate, and test the model. The code applies data augmentation techniques ๐Ÿงช to improve the robustness of the model. Transfer learning ๐Ÿ“œ is used to leverage the pre-trained weights of the Mobilenet V2 model, which significantly reduces the training time and increases the accuracy of the model. The trained model is saved in a .h5 file for future use. The code also includes a script for making predictions on new images using the trained model.

To run the code, you will need to have TensorFlow, Keras, and TensorFlow Datasets installed. You can easily customize the code to work with your own dataset by modifying the data loading and preprocessing steps. The code is well-commented ๐Ÿ’ฌ and should be easy to understand even if you are new to TensorFlow and Keras.

mobilenet_v2's People

Contributors

liyaunknown avatar

Stargazers

 avatar  avatar

Watchers

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