Ivy is an open-source video-based object counting software for tallying pretty much anything (vehicles, people, animals — you name it).
- Python 3 (tested with versions 3.5, 3.6 and 3.7)
- Clone this repo
[email protected]:nicholaskajoh/ivy.git
. - Create and/or use a virtual environment (optional but recommended).
- Install the dependencies in requirements.txt
pip install -r requirements.txt
. - Choose a detector and install its dependencies where necessary (if you're not sure what to pick, we recommend you start with
yolo
).
Detector | Description | Dependencies |
---|---|---|
yolo |
Perform detection using models created with the YOLO (You Only Look Once) neural net. https://pjreddie.com/darknet/yolo/ | |
tfoda |
Perform detection using models created with the Tensorflow Object Detection API. https://github.com/tensorflow/models/tree/master/research/object_detection | CPU: pip install tensorflow-cpu GPU: pip install tensorflow-gpu |
detectron2 |
Perform detection using models created with FAIR's Detectron2 framework. https://github.com/facebookresearch/detectron2 | python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' (https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md) |
haarcascade |
Perform detection using Haar feature-based cascade classifiers. https://docs.opencv.org/3.4/db/d28/tutorial_cascade_classifier.html |
- Create a .env file (based on .env.example) in the project's root directory and edit as appropriate.
- Run
python -m main
.
Download ivy_demo_data.zip and unzip its contents in the data directory. It contains detection models and a sample video.
python -m pytest
By default, Ivy runs in "debug mode" which provides you a window to monitor the object counting process. You can:
- press the
p
key to pause/play the counting process - press the
s
key to capture a screenshot - press the
q
key to quit the program - click any point on the window to log the coordinates of the pixel in that position
Got questions, contributions, suggestions, concerns? Join our community on Spectrum! Also follow us on Twitter @CountWithIvy to get notified about new features, fixes and initiatives.