qqwweee/keras-yolo3
Keras Yolo3
This project is an object detection framework implementing the YOLOv3 architecture using Keras and TensorFlow. It functions as a deep learning vision model and computer vision toolset designed to locate and classify multiple entities within images and video streams using bounding boxes.
The system includes a multi-GPU inference engine to distribute computational loads across several graphics processing units. It also provides a pipeline for creating custom object detectors by retraining pre-trained weights on annotated datasets to recognize user-defined object classes.
The framework covers model training and fine-tuning through a two-stage retraining process and weight optimization. It includes utilities for network architecture configuration via external files, weight format conversion between frameworks, and the transformation of VOC annotations into plain text for training.
The project supports inference across static images, live streams, and sequential video files.
Features
- YOLOv3 Implementations - Implements the YOLOv3 architecture using Keras and TensorFlow for real-time image and video object detection.
- Object Detection - Identifies and locates multiple objects within static images using bounding boxes and classification.
- Video Stream Detections - Processes sequential video frames to detect objects and exports the resulting detection output.
- Real-Time Object Detection - Provides real-time identification and localization of objects within live images and video streams.
- Detection Model Training - Allows training object detection models on custom annotated datasets using pre-trained weights.
- Deep Learning Inference Engines - Executes pretrained neural networks on visual data to generate bounding boxes and class labels.
- Keras Model Implementations - Implements the YOLOv3 architecture specifically using the high-level Keras API.
- Deep Learning Models - Implements a deep learning vision model for locating and classifying multiple entities using bounding boxes.
- Vision Model Fine-Tuning - Optimizes the final layers of pre-trained vision models to recognize specific custom object classes.
- Computer Vision Training Frameworks - Provides a toolset for training, fine-tuning, and deploying convolutional neural networks for visual recognition.
- Vision Model Retraining Pipelines - Employs a two-stage retraining process that first stabilizes frozen layers before training the entire network.
- Multi-GPU Inference Runtimes - Distributes the computational load of deep learning predictions across multiple graphics processing units.
- Multi-GPU Workload Distribution - Distributes the computational load across multiple GPUs to accelerate object detection processing speed.
- Two-Stage Weight Stabilization - Optimizes the model through a two-stage retraining process that stabilizes early layers before full-network fine-tuning.
- Video Analysis - Processes video files to automatically detect and track entities across frames for monitoring or tagging.
- Custom YOLO Detectors - Provides a pipeline for retraining pre-trained YOLO weights on custom datasets to recognize user-defined object classes.
- Model Architecture Configurations - Allows defining the neural network structure using external configuration files instead of hard-coded layers.
- Computer Vision Libraries - Keras implementation of the YOLOv3 object detector.
