YOLOv4 implementation with Tensorflow 2
Project description
tf-yolov4
YOLOv4 implementation with Tensorflow 2.
Install
pip instal tf-yolov4
Example
Prediction
import numpy as np
import PIL.Image
import yolov4
# Default: num_classes=80
yo = yolov4.YOLOv4(num_classes=80)
# Default: weights_path=None
# num_classes=80 and weights_path=None: Pre-trained COCO model will be loaded.
# num_classes!=80 and weights_path=None: Pre-trained backbone and SPP model will be loaded.
# Otherwise: User-defined weight file will be loaded.
yo.load_weights(weights_path=None)
img = np.array(PIL.Image.open('./data/sf.jpg'))
# The image with predicted bounding-boxes is created if `debug=True`
boxes, classes, scores = yo.predict(img, debug=True)
Load Darknet weight
import yolov4
yo = yolov4.YOLOv4(num_classes=10)
yo.load_darknet_weights('/path/to/darknet_weight')
TODO
- Prediction
- Load Darknet weight file
- Pre-trained model
- Basic training function and Loss definition
- Label-smoothed BCE loss
- c-IoU loss
- Training data augmentation
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tf-yolov4-1.0.4.tar.gz
(11.0 kB
view hashes)
Built Distribution
tf_yolov4-1.0.4-py3-none-any.whl
(21.6 kB
view hashes)
Close
Hashes for tf_yolov4-1.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b0694fa134429e3b06e74bdff244b8d82ba9716a11fed71e6436c6a677a24036 |
|
MD5 | 01010386f53a47ab938af0a8e4d2d1c4 |
|
BLAKE2b-256 | 13067497758c412029d6f67b12beda3437259316fc1a11780705bafdbca7d771 |