Skip to content
/ CNN Public

Adopted a combination of [AutoAugment + Transfer Learning + Ensemble Model] approach to solve small dataset classification problem in CNN.

Notifications You must be signed in to change notification settings

Jingshun99/CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CNN

This project introduces possible solutions that could be implemented to increase the accuracy of Convolutional Neural Network(CNN) with a small sample size.

Environment:

  • Windows 10 64-bit
  • Python 3.9.7
  • Visual Studio code

To start, run main.ipynb using Jupyter notebook or Google Colab.

Table of Content decsribes what each cell is doing.

To make prediction on own image, add your image into the image subfolder, and replace the <img_pth> in cell 27 with your image path.

All the trained model will be saved in the weights subfolder and the output figures will be saved in the output subfolder.

About

Adopted a combination of [AutoAugment + Transfer Learning + Ensemble Model] approach to solve small dataset classification problem in CNN.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published