No Code TF Image Classification Trainer

Start Training a State of the Art Image Classifier within Minutes with No Coding Knowledge
Demo Video · Docker Image · Report Bug · Request Feature

Table of Contents

• Table of Contents
• About The Project
• Demo
• Features
• Hardware Requirements
  • Hardware used for Development and Testing
  • Minimum Hardware Requirements
• Getting Started
  • Prerequisites
  • Built With
• Setup and Usage
• Changelog
• Roadmap
• Contributing
• License
• Contact
  • Animikh Aich

About The Project

Containerized Tensorflow-based image classification training utility with Streamlit-based interface designed to choose between common architectures and optimizers for quick hyperparameter tuning, which drastically lowers experimentation time.

Demo

YouTube Video Link: https://youtu.be/gbuweKMOucc

Features

• No Coding Required - I have said this enough, I will repeat one last time: No need to touch any programming language, just a few clicks and start training!
• Easy to use UI Interface - Built with Streamlit, it is a very user friendly, straight forward UI that anybody can use with ease. Just a few selects and a few sliders, and start training. Simple!
• Live and Interactive Plots - Want to know how your training is progressing? Easy! Visualize and compare the results live, on your dashboard and watch the exponentially decaying loss curve build up from scratch!

If you want to go in-depth with the Technical Details, then there are too many to list here. I would invite you to check out the Changelog where every feature is mentioned in details.

Hardware Requirements

We recommend an Nvidia GPU for Training, However, it can work with CPUs as well (Not Recommended)

Google Cloud TPUs are Supported as per the code, however, the same has not been tested.

Hardware used for Development and Testing

• CPU: AMD Ryzen 7 3700X - 8 Cores 16 Threads
• GPU: Nvidia GeForce RTX 2080 Ti 11 GB
• RAM: 32 GB DDR4 @ 3200 MHz
• Storage: 1 TB NVMe SSD
• OS: Ubuntu 20.10

The above is just used for development and by no means is necessary to run this application. The Minimum Hardware Requirements are given in the next section

Minimum Hardware Requirements

• CPU: AMD/Intel 4 Core CPU (Intel Core i3 4th Gen or better)
• GPU: Nvidia GeForce GTX 1650 4 GB (You can go lower, but I would not recommend it)
• RAM: 8 GB (Recommended 16 GB)
• Storage: Whatever is required for Dataset Storage + 10 GB for Docker Image
• OS: Any Linux Distribution

Getting Started

Prerequisites

• Docker
• Nvidia Docker
• Python 3
• Git

Built With

• Streamlit
• Tensorflow

Setup and Usage

1. Install Docker Engine
2. Install Nvidia Docker Engine (Required only for System with Nvidia GPU)
3. Set up the Dataset Structure:

.
├── Training
│   ├── class_name_1
│   │   └── *.jpg
│   ├── class_name_2
│   │   └── *.jpg
│   ├── class_name_3
│   │   └── *.jpg
│   └── class_name_4
│       └── *.jpg
└── Validation
    ├── class_name_1
    │   └── *.jpg
    ├── class_name_2
    │   └── *.jpg
    ├── class_name_3
    │   └── *.jpg
    └── class_name_4
        └── *.jpg

4. There are two methods:
   • Pull the Image from Docker Hub Repository
   • Clone the Repo and Build the Docker Image using the build script
5. Once the Docker Image is Built/Pulled, run the image using the following command:

docker run -it --runtime nvidia --net host -v /path/to/dataset:/data <image-name>

6. After training the trained weights can be found at: /app/model/weights Inside the Container
7. After training the Tensorboard Logs can be found at: /app/logs/tensorboard Inside the Container
8. You can use docker cp <container-name/id>:<path-inside-container> <path-on-host-machine> to get the weights and logs out. Further details can be found here: Docker cp Docs

Changelog

See the Changelog.

Roadmap

See the Open Issues for a list of proposed features (and known issues).

See the Changelog a lost of changes currently in development.

Contributing

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

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

License

Distributed under the GNU AGPL V3 License. See LICENSE for more information.

Contact

Animikh Aich

• Website: Animikh Aich - Website
• LinkedIn: animikh-aich
• Email: [email protected]
• Twitter: @AichAnimikh