Skip to content

ariharasudhanm/Image-classification-using-transfer-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Contributors Last-commit MIT License LinkedIn


Image classification using transfer learning

Classification of acoustic and electric guitar with transfer learning methods using tensor flow
Explore the docs »

· Report Bug · Request Feature

Table of Contents
  1. About The Project
  2. Usage
  3. Roadmap
  4. Results
  5. Contributing
  6. License
  7. Contact
  8. Acknowledgments

About The Project

Training samples | Sample data augmentation | Training and validation accuracy

Project Overview:

  • Dataset creation(train, validation, test splits) including data augmentation.
  • Training few layers layers and perform validation to know whether to further train few more layers.
  • Validation and testing.

(back to top)

Built With

These are programming languages, libraries, frameworks and other tools used in this project.

(back to top)

Usage

  • There are three directories inside the dataset directory called test, val(for validation) and train where each directories contains 2 directories called acoustic and electric which contains respective categories of images (this is done manually to avoid API contradictions in tensorflow but it is better to split it randomly in order to avoid bias in train,validation,test set splitting process).
  • In the notebook we use absolute path of these directories to feed it when creating a train, test, validation datasets(be aware of this while running).
  • In order to avoid confusions running it on different OS platforms we are considering the absloute path of the images while creating datasets.
  • You can add more images to their respective images directories if needed.

Roadmap

  • Data split and data augmentation
  • Training and validation
  • Testing

See the open issues for a full list of proposed features (and known issues).

(back to top)

Results

Testing the trained model for first six images from test set Images/Test_Samples.png.

Testing_few_images

Contributing

Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.

If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again!

  1. Fork the Project
  2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
  3. Commit your Changes (git commit -m 'Add some AmazingFeature')
  4. Push to the Branch (git push origin feature/AmazingFeature)
  5. Open a Pull Request

(back to top)

License

Distributed under the MIT License. See LICENSE.txt for more information.

(back to top)

Contact

Your Name - @AriharasudhanM - ariharasudhan.muthusami@gmail.com

Project Link: Image-classification-using-transfer-learning

(back to top)

Acknowledgments

Use this space to list resources you find helpful and would like to give credit to. I've included a few of my favorites to kick things off!

(back to top)

About

Image classfication using transfer learning with tensorflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published