YOLOv3 implementation in TensorFlow 2.x
Project description
YOLOv3-TF
YOLOv3 implementation in TensorFlow 2.x
Installation
pip install yolov3-tf
Depends on tensorflow >=2.3.0 <=2.9.1
Usage
The package consists of three core modules -
- dataset
- models
- utils
Dataset
The dataset.py
module is for loading and transforming the tfrecords for object detection. The examples in the input tfrecords must match the parsing schema.
import yolov3_tf.dataset as dataset
train_dataset = dataset.load_tfrecord_dataset(tfrecords_path)
train_dataset = train_dataset.batch(batch_size)
train_dataset = train_dataset.map(
lambda x, y: (
dataset.transform_images(x, image_dim),
dataset.transform_targets(y, anchors, anchor_masks, image_dim),
)
)
Models
The models.py
module consists of implementation of two YOLOv3 and YOLOv3 tiny in Tesnsorflow.
from yolov3_tf.models import YoloV3, YoloV3Tiny
model = YoloV3(image_dim = 416, training=True, classes=10)
Utils
The utils.py
module provides some common functions for training YOLOv3 model, viz., loading weights, freezing layers, drawing boxes on images, compute iou
# convert weights
from yolov3_tf.models import YoloV3, YoloV3Tiny
from yolov3_tf import utils
yolo = YoloV3()
utils.load_darknet_weights(yolo, weights_path, is_tiny=False)
yolo.save_weights(converted_weights_path)
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
yolov3-tf-0.1.3.tar.gz
(11.7 kB
view hashes)
Built Distribution
yolov3_tf-0.1.3-py3-none-any.whl
(12.8 kB
view hashes)
Close
Hashes for yolov3_tf-0.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 251dcc4aae0174603f8da097b7576be2de62c21ae50ad32f5806e4926234bf5e |
|
MD5 | be55bf269620ebe951801bc8a2565fe4 |
|
BLAKE2b-256 | 715c256496fadadee0529993f7480e715da4f6cb713da1a482a82a28209ed62b |