Official English Documentation for ImageAI!

ImageAI is a python library built to empower developers, reseachers and students to build applications and systems with self-contained Deep Learning and Computer Vision capabilities using simple and few lines of code. This documentation is provided to provide detailed insight into all the classes and functions available in ImageAI, coupled with a number of code examples. ImageAI is a project developed by Moses Olafenwa.

The Official GitHub Repository of ImageAI is https://github.com/OlafenwaMoses/ImageAI

ImageAI now uses PyTorch backend.

For full details on this and if you plan on using existing Tensorflow pretrained models, custom models and Pascal VOC dataset, visit the BACKEND_MIGRATION.md documentation.

_images/image11.jpg

Installing ImageAI

ImageAI requires that you have Python 3.7.6 installed as well as some other Python libraries and frameworks. Before you install ImageAI, you must install the following dependencies.

  • Download and Install Python 3.7, Python 3.8, Python 3.9 or Python 3.10

  • Install Dependencies (CPU)

    pip install cython pillow>=7.0.0 numpy>=1.18.1 opencv-python>=4.1.2 torch>=1.9.0 --extra-index-url https://download.pytorch.org/whl/cpu torchvision>=0.10.0 --extra-index-url https://download.pytorch.org/whl/cpu pytest==7.1.3 tqdm==4.64.1 scipy>=1.7.3 matplotlib>=3.4.3 mock==4.0.3
    
  • Install Dependencies (GPU/CUDA)

    pip install cython pillow>=7.0.0 numpy>=1.18.1 opencv-python>=4.1.2 torch>=1.9.0 --extra-index-url https://download.pytorch.org/whl/cu102 torchvision>=0.10.0 --extra-index-url https://download.pytorch.org/whl/cu102 pytest==7.1.3 tqdm==4.64.1 scipy>=1.7.3 matplotlib>=3.4.3 mock==4.0.3
    
  • If you plan to train custom AI models, run the command below to install an extra dependency

    pip install pycocotools@git+https://github.com/gautamchitnis/cocoapi.git@cocodataset-master#subdirectory=PythonAPI
    
  • ImageAI

    pip install imageai --upgrade
    

Once ImageAI is installed, you can start running very few lines of code to perform very powerful computer visions tasks as seen below.

Image Recognition

Recognize 1000 different objects in images

_images/image2.jpg
  • convertible : 52.459555864334106
  • sports_car : 37.61284649372101
  • pickup : 3.1751200556755066
  • car_wheel : 1.817505806684494
  • minivan : 1.7487050965428352

Visit Documentation

Image Object Detection

Detect 80 most common everyday objects in images.

_images/image3.jpg

Visit Documentation

Video Object Detection

Detect 80 most common everyday objects in videos.

_images/image4.gif

Visit Documentation

Video Detection Analysis

Generate time based analysis of objects detected in videos.

_images/image5.gif

Visit Documentation

Custom Image Recognition Training and Inference

Train new image new deep learning models on recognize custom objects

_images/image61.jpg

Visit Documentation

Custom Objects Detection Training and Inference

Train new YOLOv3 models to detect custom objects

_images/image71.jpg

Visit Documentation

Follow the links in the Content section below to see all the code samples and full documentation of available classes and functions.

Indices and tables