Skip to content

Latest commit

 

History

History
142 lines (78 loc) · 4.36 KB

index.rst

File metadata and controls

142 lines (78 loc) · 4.36 KB
Raw

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.

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

  • 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.

Visit Documentation

Video Object Detection

Detect 80 most common everyday objects in videos.

Visit Documentation

Video Detection Analysis

Generate time based analysis of objects detected in videos.

Visit Documentation

Custom Image Recognition Training and Inference

Train new image new deep learning models on recognize custom objects

Visit Documentation

Custom Objects Detection Training and Inference

Train new YOLOv3 models to detect custom objects

Visit Documentation

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

prediction/index detection/index video/index custom/index customdetection/index

Indices and tables

  • genindex
  • modindex
  • search