Skip to content

This is a project about the work of a transfer learning (from binary image classification) to multi category image classification and regression, as well as some data processing, regularization, and data augmentation.

Notifications You must be signed in to change notification settings

kmock930/Multi-Category-Image-Modelling-Transfer-Learning

Repository files navigation

Computer Vision based Transfer Learning from this project: https://github.com/kmock930/Texture-Image-Comparison.git

Project Preview

Trained Bounding Box to identify an object in an image Sample Image BEFORE processing Sample Image AFTER processing

Aims

  • Build 2 neural-network-based classifiers (i.e., ResNet50 and ResNet50-V2) to recognize more categories in the stonefiles dataset: https://web.engr.oregonstate.edu/~tgd/bugid/stonefly9/
  • Perform Data Preprocessing.
  • Evaluate the model's performance via some standard metrics and predictions.
  • Perform Transfer Learning to learn a bounding box which accurately circles an object in an image.
  • Perform Multi-Task Regularization and Data Augmentation.
  • Evaluate the Transfer Learning process.

Project Prerequisites

  • Run the command pip freeze > requirements.txt to generate the latest version of requirements.txt file which keeps track of all necessary pip installs.

Project Structure

  • config.yaml is a configuration file that stores all parameters for modelling, training, experimenting, evaluating and etc.
  • poc_hydra.py is a standalone script to test the interaction with hydra in order to access (read/write) parameters in config.
  • Check out my Jupyter Notebook to see my analysis.

About

This is a project about the work of a transfer learning (from binary image classification) to multi category image classification and regression, as well as some data processing, regularization, and data augmentation.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published