Classification of acoustic and electric guitar with transfer learning methods using tensor flow
Explore the docs »
·
Report Bug
·
Request Feature
Table of Contents
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.
These are programming languages, libraries, frameworks and other tools used in this project.
- 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.
- Data split and data augmentation
- Training and validation
- Testing
See the open issues for a full list of proposed features (and known issues).
Testing the trained model for first six images from test set Images/Test_Samples.png
.
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!
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature'
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request
Distributed under the MIT License. See LICENSE.txt
for more information.
Your Name - @AriharasudhanM - ariharasudhan.muthusami@gmail.com
Project Link: Image-classification-using-transfer-learning
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!
- Tensor flow transfer learning documentation
- Tensor flow transfer learning tutorials
- Transfer Learning Guide: A Practical Tutorial With Examples for Images and Text in Keras
- Transfer Learning with TensorFlow Tutorial: Image Classification Example
- Transfer learning for Deep Neural Networks using TensorFlow
- Transfer learning and fine-tuning in Keras and Tensorflow