Skip to content

Predict whether images are 'happy' or 'sad'. Part of machine learning module coursework.

License

Notifications You must be signed in to change notification settings

crong12/image-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification Project

Project Overview

This project is the main coursework for a machine learning module I took in university. The task was image classification with a binary output - happy vs sad.

This readme file will not contain the full details of the project - these may be found in the project report.

The final grade received for this submission was 94/100.

Data

The training and test datasets are located within the data folder. However, the labels for the test data are not available as these are withheld by the course instructor.

The training dataset contains a total of 2,500 observations and 2,304 features. Of these, 2,048 features were pre-extracted from the fc7 layer of the CaffeNet CNN, while the remaining 256 features were GIST features. confidence labels were included to indicate the confidence level that each image was correctly labeled. When all 3 labellers were in agreement of an image's classification, the confidence was 1. Otherwise, it was 0.66. The test dataset contains only the 2,048 CaffeNet CNN features and the 256 GIST features.

Output

The output of this project was an array of label predictions for the test data, which may be found in the output folder.

License

This project is licensed under the MIT License. You are free to use the code and resources for educational or personal purposes.

Contact

Please feel free to contact me regarding any questions or feedback regarding this project. I can be reached at ongcrong@yahoo.com.

Thank you for taking the time to look at this project!

About

Predict whether images are 'happy' or 'sad'. Part of machine learning module coursework.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published