ashivdhondea/ensc813_project

ENSC 813 Deep Learning Systems in Engineering - Term Project

ENSC813_Project

ENSC 813 Deep Learning Sytems in Engineering - Term Project

Classifying car images in the TCC Dataset

This project implements ConvNets for binary and multi-class classification of car images in the The Car Connection dataset.

Credits

The The Car Connection dataset was scraped by Mr. Nicolas Gervais. The provision of this dataset for free is gratefully acknowledged. The source repo for this dataset is Predicting car price from scraped data.

• Hans Dhondea - This repo - AshivDhondea

Report

The project report and its source files can be found in this repo.

User manual

A brief user manual for this code can be found in the user manual directory of the report repo.

Learned models

Due to GitHub's file size restrictions, most models cannot be committed to this repo. The model and weights files created by main_22_multiclass_00.py can be found in the output_files directory.

Prerequisites

This project was created using a custom conda environment called tfgpu.

To re-create this environment, the following command may be run in the Anaconda Powershell Prompt: conda create -n tfgpu --ensc813-tfgpu-package-list.txt . More info in the requirements directory.

Python packages used are listed in the file ensc813-tfgpu-package-list.txt which can be obtained by running the command conda list --export > ensc813-tfgpu-package-list.txt .

Binary classification

For the binary classification task, we made use of the following ConvNet architectures

Multi-class classification

We modified the binary classification ConvNets to accommodate multi-class classification problems. We ensembled our ConvNets to obtain an improved multi-class classifier. The following confusion matrix summarizes its classification performance.

Citation

If you use this work, cite it as

@misc{Hans2020DL,

author = {Dhondea, A.R.},

title = {Classifying car images in the TCC Dataset},

year = {2020},

publisher = {GitHub},

journal = {GitHub repository},

howpublished = {\url{https://github.com/AshivDhondea/ENSC813_Project}}

}

License

This project is licensed under the MIT License - see the LICENSE.md file for details