yolov4 0.5.0

No project description provided

tensorflow-yolov4

python3 -m pip install yolov4

YOLOv4 Implemented in Tensorflow 2. Convert YOLOv4, YOLOv3, YOLO tiny .weights to .pb, .tflite and trt format for tensorflow, tensorflow lite, tensorRT.

Download yolov4.weights file: https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

Performance

tensorflow

import yolov4.tf as yolo

detector = yolo.YoloV4(
    names_path="/home/hhk7734/tensorflow-yolov4/data/classes/coco.names",
    weights_path="/home/hhk7734/Desktop/yolov4.weights",
)

detector.inference(media_path="/home/hhk7734/tensorflow-yolov4/data/kite.jpg",)

tensorflow lite

import yolov4.tflite as yolo

detector = yolo.YoloV4(
    names_path="/home/hhk7734/tensorflow-yolov4/data/classes/coco.names",
    tflite_path="/home/hhk7734/Desktop/yolov4.tflite",
)

detector.inference(
    media_path="/home/hhk7734/tensorflow-yolov4/data/road.mp4",
    is_image=False,
    video_interval_ms=1000,
)

Training

import yolov4.tf as yolo

detector = yolo.YoloV4(
    names_path="/home/hhk7734/tensorflow-yolov4/data/classes/coco.names"
)

detector.train(
    train_annote_path="/home/hhk7734/tensorflow-yolov4/data/dataset/val2017.txt",
    test_annote_path="/home/hhk7734/tensorflow-yolov4/data/dataset/val2017.txt",
    pre_trained_weights="/home/hhk7734/Desktop/yolov4.weights",
    epochs=2000,
    save_interval=100,
)

import yolov4.tf as yolo

detector = yolo.YoloV4(
    names_path="/home/hhk7734/darknet/data/class.names"
)

detector.train(
    train_annote_path="/home/hhk7734/darknet/data/train.txt",
    test_annote_path="/home/hhk7734/darknet/data/train.txt",
    dataset_type="yolo",
    epochs=2000,
    save_interval=100,
)

tensorflow-yolov4 (0.5.0) unstable; urgency=medium

• core: dataset: add yolo type
• tf: add dataset_type parameter to YoloV4.train
• tf: add epochs parameter to train
• tf: add save_interval parameter to train

-- Hyeonki Hong [e-mail] Fri, 19 Jun 2020 14:30:50 +0900

tensorflow-yolov4 (0.4.0) unstable; urgency=medium

• core: dataset: remove cfg module
• tf: implement YoloV4.train

-- Hyeonki Hong [e-mail] Thu, 11 Jun 2020 17:45:44 +0900

tensorflow-yolov4 (0.3.0) unstable; urgency=medium

• core: utils: use numpy instead of tensorflow
• pypi: remove install_requires and change to manual installation
• yolov4: add video_interval_ms

-- Hyeonki Hong [e-mail] Mon, 08 Jun 2020 23:59:41 +0900

tensorflow-yolov4 (0.2.0) unstable; urgency=medium

• pylint: create .pylintrc and run black
• core: remove config.py
• yolov4: change tfyolov4 to yolov4
• yolov4: remove detect**.py and implement YoloV4.inference

-- Hyeonki Hong [e-mail] Mon, 08 Jun 2020 02:20:49 +0900

tensorflow-yolov4 (0.1.0) unstable; urgency=medium

• yolov4: fork from 'hunglc007/tensorflow-yolov4-tflite'

-- Hyeonki Hong [e-mail] Fri, 05 Jun 2020 20:17:45 +0900