#109 · Primary category: Computer Vision

ImageAI

ai-practice-recommendations algorithm artificial-intelligence artificial-neural-networks densenet detection gpu image-prediction image-recognition imageai inceptionv3 machine-learning object-detection offline-capable prediction python python3 squeezenet video

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities

Project last updated:08/03/24

GitHub Stars

8.9K

Forks

2.2K

Contributors

19

License

MIT

Why we included this project

ImageAI is a Python library that keeps computer vision approachable: a few lines of code get you image classification, object detection, and the same detection pipeline running on video with objects tracked across frames. It wraps pretrained models such as MobileNetV2, ResNet50, and YOLOv3 and handles the weights for you, so you can skip the model-serving code and focus on the result, which makes it a good fit for prototypes and internal tools. You can also train custom detection and recognition models on your own object classes instead of staying stuck with the bundled datasets. Small teams that want a working vision pipeline without heavy infrastructure can get surprisingly far with it before needing something bigger.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category