yolov3  by ultralytics

Object detection in PyTorch > ONNX > CoreML > TFLite

Created 7 years ago
10,453 stars

Top 4.9% on SourcePulse

GitHubView on GitHub
Project Summary

Ultralytics YOLOv3 provides a PyTorch implementation of the YOLOv3 object detection model, optimized for speed and accuracy. It targets researchers and developers needing a robust baseline for computer vision tasks, offering straightforward integration and deployment across multiple platforms.

How It Works

This implementation leverages PyTorch, building upon the established YOLOv3 architecture. It utilizes Darknet-53 as its feature extractor and incorporates multi-scale predictions and logistic classifiers for improved accuracy, particularly with small objects and multi-label classification. The project emphasizes ease of use and exportability to formats like ONNX, CoreML, and TFLite.

Quick Start & Requirements

  • Install via pip: pip install ultralytics
  • Clone repo and install dependencies: git clone https://github.com/ultralytics/yolov3, cd yolov3, pip install -r requirements.txt
  • Requires Python >= 3.8.0 and PyTorch >= 1.8.
  • Inference can be performed using PyTorch Hub or the detect.py script.
  • Training examples are provided for COCO dataset.
  • Official Docs: https://docs.ultralytics.com/

Highlighted Details

  • Supports inference via PyTorch Hub and detect.py script with various input sources (images, videos, streams, webcam).
  • Offers training scripts for YOLOv3, YOLOv3-SPP, and YOLOv3-tiny on datasets like COCO.
  • Includes tutorials for custom data training, model export (ONNX, TFLite, CoreML), and deployment on NVIDIA Jetson.
  • Integrates with experiment tracking tools like Weights & Biases and Comet ML.

Maintenance & Community

Licensing & Compatibility

  • AGPL-3.0 License: Suitable for academic and personal projects, requires derivative works to be open-sourced.
  • Enterprise License available for commercial use.

Limitations & Caveats

  • While the repo is named yolov3, the README indicates it's based on YOLOv5 structure, suggesting potential compatibility nuances for pure YOLOv3 users.
  • YOLOv3-specific documentation may be limited, directing users to general YOLO principles.
Health Check
Last Commit

3 weeks ago

Responsiveness

1 day

Pull Requests (30d)
4
Issues (30d)
0
Star History
24 stars in the last 30 days

Explore Similar Projects

Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Edward Sun Edward Sun(Research Scientist at Meta Superintelligence Lab), and
5 more.

attorch by BobMcDear

0.2%
576
PyTorch nn module subset, implemented in Python using Triton
Created 2 years ago
Updated 1 month ago
Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems") and Kevin Hou Kevin Hou(Head of Product Engineering at Windsurf).

ImageAI by OlafenwaMoses

0.0%
9k
Python library for computer vision tasks
Created 7 years ago
Updated 1 year ago
Feedback? Help us improve.