Visual_UI
Visual UI interface for fastai
Aim to provide an easy to use graphical interface without the need to dig deep into the code. This visual tool provides a good starting point to get training quickly using fastai.
All tabs are provided within an accordion design using ipywidgets, this allows for all aspects of choosing and viewing parameters in one line of sight
The Augmentation tab utilizes fastai parameters so you can view what different image augmentations look like and compare
View batch information
Review model data and choose suitable metrics for training
Review parameters get learning rate and train using the one cycle policy
Can experiment with various learning rates and train
Requirements
- fastai
I am using the developer version:
git clone https://github.com/fastai/fastai
cd fastai
tools/run-after-git-clone
pip install -e ".[dev]"
for installation instructions visit Fastai Installation Readme
- ipywidgets
pip install ipywidgets jupyter nbextension enable --py widgetsnbextension
or
conda install -c conda-forge ipywidgets
for installation instructions visit Installation docs
- psutil
psutil (process and system utilities) is a cross-platform library for retrieving information on running processes and system utilization (CPU, memory, disks, network, sensors) in Python
pip install psutil
Installation
git clone this repository
gitclone https://github.com/asvcode/Vision_UI.git
run 01_UI_Fastai.ipynb ,specify your path and run display_ui(path)
Known Issues
Google Colab
Colab does not currently support ipywidgets because their output is in its own frame so prevents ipywidets from working. This is the link to issues thread Link
Future Work
- Currently on works with images using the
ImageDataBunch.from_folderoption. Plans to expand to.from_csvand.from_df
