ImageClassification

This repository is used for image classification task and provides training, prediction, prediction GUI, prediction Web GUI, hyperparameters tuning, and pretrained models.

Installation

pip install -r requirements.txt

Prepare dataset

Please prepare the dataset according to the following examples.

dataset
├── train   #for training
│   ├── class1
│   │   ├── image1.png
│   │   ├── image2.png
│   │   └── image3.png
│   ├── class2
│   │   ├── image1.png
│   │   ├── image2.png
│   │   └── image3.png
│   └── class3
│       ├── image1.png
│       ├── image2.png
│       └── image3.png
├── val     #for validation
│   ├── class1
│   │   ├── image1.png
│   │   ├── image2.png
│   │   └── image3.png
│   ├── class2
│   │   ├── image1.png
│   │   ├── image2.png
│   │   └── image3.png
│   └── class3
│       ├── image1.png
│       ├── image2.png
│       └── image3.png
└── test     #for testing
    ├── class1
    │   ├── image1.png
    │   ├── image2.png
    │   └── image3.png
    ├── class2
    │   ├── image1.png
    │   ├── image2.png
    │   └── image3.png
    └── class3
        ├── image1.png
        ├── image2.png
        └── image3.png

Configuration

This repository provides default configuration which are MNIST and CIFAR10.

All parameters are in the YAML file.

Argparse

You can override parameters by argparse while running.

python main.py --config config.yaml --str_kwargs mode=train #override mode as 100
python main.py --config config.yaml --num_kwargs max_epochs=100 #override training iteration as 100
python main.py --config config.yaml --bool_kwargs early_stopping=False #override early_stopping as False
python main.py --config config.yaml --str_list_kwargs classes=1,2,3 #override classes as 1,2,3
python main.py --config config.yaml --dont_check #don't check configuration

Training

python main.py --config config.yml --str_kwargs mode=train # or you can set train as the value of mode in configuration

Predict

python main.py --config config.yml --str_kwargs mode=predict,root=FILE # predict a file
python main.py --config config.yml --str_kwargs mode=predict,root=DIRECTORY # predict files in the folder

Predict GUI

python main.py --config config.yml --str_kwargs mode=predict_gui    # will create a tkinter window
python main.py --config config.yml --str_kwargs mode=predict_gui --bool_kwargs web_interface=True   #will create a web interface by Gradio

Tuning

python main.py --config config.yaml --str_kwargs mode=tuning    #the hyperparameter space is in the configuration

Pretrained

This repository provides pretrained model. Please look at the pretrained directory.