A TensorFlow implementation of YOLO for real-time object detection, supporting weight conversion, training, and mobile deployment.
Darkflow is a TensorFlow port of the YOLO (You Only Look Once) real-time object detection system. It enables users to load pre-trained YOLO weights from the Darknet framework, retrain or fine-tune models with custom datasets, and export optimized graphs for mobile deployment. The project bridges the gap between Darknet's YOLO implementations and the TensorFlow ecosystem.
Machine learning engineers and researchers who want to use YOLO's real-time object detection capabilities within TensorFlow for tasks like custom model training, fine-tuning, and mobile deployment. It's also suitable for developers needing to convert Darknet-trained YOLO models to TensorFlow format.
Darkflow provides a streamlined workflow for transitioning YOLO models from Darknet to TensorFlow, offering unique features like weight conversion, JSON output for detections, and protobuf export for mobile deployment. It simplifies custom training and fine-tuning with configuration files, making YOLO accessible within the TensorFlow toolchain without rewriting models from scratch.
Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Directly loads pre-trained YOLO weights from Darknet into TensorFlow, enabling quick model porting without retraining from scratch, as shown in the --load option examples.
Supports training and fine-tuning on custom datasets with adjustable .cfg files and labels.txt, detailed in step-by-step guides for modifying architectures like tiny-yolo-voc.
Exports models to protobuf files (.pb) with metadata for easy integration into iOS and Android apps, referenced in the TensorFlow mobile demo and --savepb command.
Provides real-time detection results in JSON format with bounding box coordinates, simplifying data handling in applications, as demonstrated in the sample output.
Relies on TensorFlow 1.0, which is deprecated and lacks features from newer versions, causing compatibility issues and potential security risks in modern environments.
Only supports YOLO v1 and v2, missing advancements in YOLOv3 and later models, which offer better accuracy and efficiency for object detection tasks.
Requires building Cython extensions and managing older dependencies like OpenCV 3, making installation error-prone and time-consuming compared to plug-and-play alternatives.