Giter VIP home page Giter VIP logo

wallacelkw / vehicle-classfication-using-deep-learning Goto Github PK

View Code? Open in Web Editor NEW
0.0 1.0 0.0 233 KB

It is a simple project where vehicle classification in malaysia for research purpose. This project will be use in deep learning apporach include efficientnet and inception

Python 34.61% Jupyter Notebook 65.39%
deep-learning efficientnet h5 image-augmentation image-augmentation-preprocessing image-classification imagescraping inception python pytorch

vehicle-classfication-using-deep-learning's Introduction

Vehicle Classification

This project is use ChatGPT create a Inception and EfficientNet's code.

In this method we train a convolutional neural network with 6 classes which contain Bus, Cars, Motorcycles, Pickup, Truck and Van in Malaysia. The model it use Inception v1 and EfficientNet b3.

Source Code Folder

Main code

Setup environment pip install requirements.txt

To run the code for training, one can use the following command:
python train.py --modelname inception

To test the model one can use the following command: python test.py --modelname inception

The Model Architecture of Inception and Efficientnet will be in efficient.py and inception.py

All the information of function will be in utils.py

EarlyStopping class will be stored in pytorchtools.py

Preprocessing File-sub folder

All preprocessing used jupyter lab to perform:

all the processing will be store in these file which include image augmentation, convert to h5 file, and train test split.

  1. Augmentation image_augmentation.ipynb
  2. Web Scraping scrap_from_website.ipynb
  3. Train, validation, test splitting train_test_split.ipynb
  4. Convert into h5 file for machine learning convert2h5.ipynb

Result sub folder

The result contain Inception and Efficient classification report, training and validation 's loss and accuracy, confusion matrix and the checkpoint file of the trained model.

Screenshot

EfficientNet

Loss and Accuracy

Classification Report of Inception

InceptionNet

Loss and Accuracy

Classification Report of Inception

vehicle-classfication-using-deep-learning's People

Contributors

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