Giter Club home page Giter Club logo

scene-recognition-model's Introduction

scene-recognition-model

A deep learning model which can recognize a scene of 10 different classes:

  • inside an airport
  • Bakery
  • Bedroom
  • Greenhouse
  • Gym
  • Kitchen
  • Operating room
  • Pool
  • Restaurant
  • Toystore

The model achieved an accuracy of 94.5% on the test data by finetuning an inception-resnet-v2 model. This was a competition on Kaggle which won first place: https://www.kaggle.com/c/fcis-cs-deeplearningcompetition/leaderboard, Team14.

Installation Guide

The project was run on google colab but if you want to run on your own device you will need to install the following:

  • cv2
  • numpy
  • tensorflow
  • matplotlib
  • glob
  • PIL
  • tqdm

Dataset

The dataset that is available on the competition was converted into tfrecord to be easier to use in training. The training data was split into training and validation sets, which can be found here. The testing data can be found here.

The training & validation data should be placed in folder: train_data and the testing data should be placed in folder: test_data.

Model

The model used to achieve the accuracy can be found here.

The files of the model should be placed in folder: models.

Runing the project

  • To train your own model you should use the file: training_inception_resnet_v2.py
  • To run the model on the validation set use the file: validation_file.py
  • To run the model on the testing data use the file: testing_file.py

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.