YOLOv4 implementation with Tensorflow 2
Project description
tf-yolov4
YOLOv4 implementation with Tensorflow 2.
Example
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)
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.1.tar.gz
(9.8 kB
view hashes)
Built Distribution
tf_yolov4-1.0.1-py3-none-any.whl
(21.1 kB
view hashes)
Close
Hashes for tf_yolov4-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e2abb11d82271373a7740189417ba919e014814e8e8d1bfccaabb61c01b57383 |
|
MD5 | daa50bcf5bbd799b0843c56e160701a3 |
|
BLAKE2b-256 | 91d14fd4056600f6f92046ed92a9c73952e2e925e47dadae203752ecf8e2779f |