A package to plot pretty bounding boxes for object detection task
Project description
plotbbox
A package to plot pretty bounding boxes for object detection task
Install
pip install plotbbox
Example Usage
import cv2
from plotbbox import plotBBox
img = cv2.imread("your_img.png")
label_table = {0: "person", 1: "car"}
color_table = {0: [0, 255, 0], 1: [0, 0, 255]} # order: b, g, r
bboxes = your_detection_algorithm(img) # Shape: (N, 5), [:, 0]: class index, [:, 1:]: xmin, ymin, xmax, ymax
for bbox in bboxes:
class_idx = bbox[0]
xmin, ymin, xmax, ymax = bbox[1:]
plotBBox(img,
xmin, ymin, xmax, ymax, color=color_table[class_idx], thickness=1,
label=label_table[class_idx]) # plot bounding box on img
cv2.imwrite("bboxes_plotted_your_img.png", img)
Read docs/example.py and run for more details!
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
plotbbox-0.1.3.tar.gz
(151.5 kB
view details)
Built Distribution
plotbbox-0.1.3-py3-none-any.whl
(151.1 kB
view details)
File details
Details for the file plotbbox-0.1.3.tar.gz
.
File metadata
- Download URL: plotbbox-0.1.3.tar.gz
- Upload date:
- Size: 151.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.7.0 requests/2.25.1 setuptools/51.0.0.post20201207 requests-toolbelt/0.9.1 tqdm/4.57.0 CPython/3.8.5
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 |
9954926bf75bf1fcf1a40983c655bee3990841d0ec83dbbcb619e2c17ce3fc11
|
|
MD5 |
71c4aa0fd0bb6b5fadfc822203839442
|
|
BLAKE2b-256 |
dcee3702c50ded7f25568d7247f8dccf81c6cf5c6a11f961933dce90b5913629
|
File details
Details for the file plotbbox-0.1.3-py3-none-any.whl
.
File metadata
- Download URL: plotbbox-0.1.3-py3-none-any.whl
- Upload date:
- Size: 151.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.3.0 pkginfo/1.7.0 requests/2.25.1 setuptools/51.0.0.post20201207 requests-toolbelt/0.9.1 tqdm/4.57.0 CPython/3.8.5
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 |
1a8d9de365cfe1829caa3e44dba887a79e0555b19d98485ba51c07ed0ef6827c
|
|
MD5 |
ca3ac420602554a8ebca66dbf8329ffc
|
|
BLAKE2b-256 |
7f31cff627ddd9658b8e8a3bdd3bdca9a95bf668fae44ca0d0073661b05f6bb2
|