Skip to main content

No project description provided

Project description

license pypi language

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",
)

tensorflow-yolov4 (0.4.0) unstable; urgency=medium

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

-- Hyeonki Hong hhk7734@gmail.com 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 hhk7734@gmail.com 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 hhk7734@gmail.com 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 hhk7734@gmail.com Fri, 05 Jun 2020 20:17:45 +0900

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

yolov4-0.4.0.tar.gz (16.7 kB view details)

Uploaded Source

File details

Details for the file yolov4-0.4.0.tar.gz.

File metadata

  • Download URL: yolov4-0.4.0.tar.gz
  • Upload date:
  • Size: 16.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.41.1 CPython/3.6.9

File hashes

Hashes for yolov4-0.4.0.tar.gz
Algorithm Hash digest
SHA256 69c9a8b80d55c953f3a182dffc353a955e6390cd1ab3ebf0bd1c02587c405028
MD5 01a0adf33d5901372b77dcccf5f922ea
BLAKE2b-256 177b24ed4fc1560127d2f940259478ce48d17b50dde77edb751bf3fe364d74d0

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page