Skip to content

christ12938/COMP9417-Project-Image-Classification

Repository files navigation

comp9417

modules included in this project:

  1. image_sorter.py - takes the provided .png training images, and combine with their class labels in .npy file, to produce a directory containing the images where each class is in their own subdirectory named accordingly
  2. cleaner.py - deletes png files in a specified directory if they meet a specified percentage of almost white pixels
  3. nn_tensorflow.py - entry point of program for tensorflow neural networks pipeline, train on provided images using specified model, and run prediction on a test directory. see -h or --help for detailed list of arguments
  4. nn_tensorflow_models.py - factory methods for creating models of various structure
  5. nn_tensorflow_train.py - methods for different ways a model may be trained and fitted
  6. nn_tensorflow_dataset.py - dataclass used in the other nn_tensorflow* modules
  7. augmentation.py - program to split and augment X train, y train into augmented X_train_split, y_train_split, X_test_split, y_test_split for training and validation
  8. nn_augmentation_model.py - seperate program to train the appened augmented data in 6. with class weights

feature classification:

1-SIFT.py - uses SIFT for feature extraction and then isomap to reduce dimensionality of data

2-clustering.py - clusters the descriptors found in 1-SIFT

3-featureVectors.py - takes clusters and descriptors and creates feature vectors for each image

4-classifier.py - uses a classifier model to classify the feature vectors

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages