CenterNet implementation by Tensorflow 2
Project description
tf-centernet
CenterNet implementation with Tensorflow 2.
Install
pip instal tf-centernet
Example
CenterNet object detection
import numpy as np
import PIL.Image
import centernet
# Default: num_classes=80
obj = centernet.ObjectDetection(num_classes=80)
# Default: weights_path=None
# num_classes=80 and weights_path=None: Pre-trained COCO model will be loaded.
# Otherwise: User-defined weight file will be loaded.
obj.load_weights(weights_path=None)
img = np.array(PIL.Image.open('./data/sf.jpg'))[..., ::-1]
# The image with predicted bounding-boxes is created if `debug=True`
boxes, classes, scores = obj.predict(img, debug=True)
TODO
- Object detection
- Pre-trained model for object detection
- Pose estimation
- Pre-trained model for pose estimation
- Training function and Loss definition
- 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-centernet-1.0.5.tar.gz
(6.6 kB
view hashes)
Built Distribution
Close
Hashes for tf_centernet-1.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9881f96c24da81e720335d71c86b9979445c3781c26b42b0e34831b60437b972 |
|
MD5 | 84a7cf849a7de6bdee50850b29540e71 |
|
BLAKE2b-256 | 33458be1a19aa4e32b1d6f9f05df3497e9724b134162ccc8cce47b1a73781926 |