Simple Object Detection

Toolkit and models for object detection.

• Free software: MIT license
• Documentation: https://simple-object-detection.readthedocs.io.

Features

• Wrapper for object detections neural networks.
• Class for store object information (center, bounding box, width, height, score, label, ...)
• Abstract class for PyTorcuh hub models easy implementation.
• Implemented neuronal networks: YOLOv5 with PyTorch.
• Image tools to draw bounding boxes on it.
• Video tools to load and easily generate detections.
• File geneneration with objects detections.
• Objects filters based on its label, score, region, etc.

Credits

This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.