A TensorFlow 2.0 implementation of YOLOv4: Optimal Speed and Accuracy of Object Detection
This implementation runs (for now) inference with the original Darknet weights from AlexeyAB. See the roadmap section to see what's next.
To install this package, you can run:
pip install https://github.com/sicara/tf2-yolov4/archive/master.zip pip install tensorflow # Check that tf2_yolov4 is installed properly python -c "from tf2_yolov4.model import YOLOv4; print(YOLOv4)"
To load the Darknet weights trained on COCO, you have to:
TF weights should be saved as
For more information about the conversion script, run