Load a pre-trained object detection model and run it on your own photos with just a few lines of Python.
Fine-tune a GluonCV model on a custom image dataset to recognize objects specific to your project.
Reproduce published computer vision research results using the included training scripts and benchmark configurations.
| dmlc/gluon-cv | opengvlab/llama-adapter | aiwaves-cn/agents | |
|---|---|---|---|
| Stars | 5,923 | 5,923 | 5,922 |
| Language | Python | Python | Python |
| Setup difficulty | moderate | hard | moderate |
| Complexity | 3/5 | 4/5 | 3/5 |
| Audience | data | researcher | developer |
Figures from each repo's GitHub metadata at analysis time.
Requires MXNet or PyTorch to be installed, a GPU significantly speeds up training and inference.
GluonCV is a Python toolkit for computer vision tasks built on top of the MXNet deep learning library, with support for PyTorch as well. Computer vision here means teaching a program to understand images and videos: recognizing what objects appear in a photo, drawing boxes around them, labeling individual pixels, tracking human body positions, or identifying actions in video clips. The toolkit ships with over 50 pre-trained models covering six main tasks: image classification, object detection, semantic segmentation, instance segmentation, human pose estimation, and video action recognition. Pre-trained means these models were already trained on large public datasets and are ready to use without starting from scratch. A developer can load one of these models with a few lines of code and start making predictions immediately. Beyond just inference, GluonCV includes the full training scripts used to produce the results reported in published research papers. This lets a researcher or engineer reproduce a known result, or adapt the training process to their own dataset. The APIs are designed to reduce the amount of setup code needed, so teams can go from a raw dataset to a trained model without writing boilerplate infrastructure. The project is maintained by the DMLC open-source community. It is released under the Apache 2.0 license and is installable via PyPI. The documentation includes tutorials covering each supported task. The README also points to AutoGluon as an alternative for users who want a more automated, lower-configuration approach to image classification and object detection with a broader range of underlying model architectures.
A Python toolkit with 50+ ready-to-use computer vision models for object detection, image segmentation, human pose estimation, and video recognition, built on MXNet with PyTorch support.
Mainly Python. The stack also includes Python, MXNet, PyTorch.
Use freely for any purpose including commercial products, keep the copyright notice and the Apache 2.0 license file.
Setup difficulty is rated moderate, with roughly 30min to a first successful run.
Mainly data.
This repo across BitVibe Labs
Verify against the repo before relying on details.