This repository contains the codes and models used in our CZ4041 Machine Learning Project. The project is on Leaf Classification Competition on Kaggle. The details of the competition and the dataset used can be found here
- Bryan Leow Xuan Zhen
- Feng Daifei
- Kong Jie Wei
- Wang Xiaoyu
- Wu Wenxuan
- Image Preprocessing
image_preprocessing.py
: Our code used to preprocess the images by adding black padding to transform the images into squares
- Models
CNN-1D.ipynb
: Code for 1-Dimensional Convolutional Neural NetworkCNN-2D Transfer Learning (MobileNet).ipynb
: Code for 2D CNN with Transfer LearningCNN-2D.ipynb
Code for 2D CNNEnsemble Model.ipynb
Code for Ensemble modelLogistic Regression.ipynb
Code for Logistic RegressionMultilayer Perception.ipynb
Code for Multilayer Perceptron
- Project Report.pdf: Our Project Report
Use Jupyter notebook and run each of the Jupyter notebook *.ipynb
file.