Skip to main content

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


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)

Uploaded Source

Built Distribution

plotbbox-0.1.3-py3-none-any.whl (151.1 kB view details)

Uploaded Python 3

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

Hashes for plotbbox-0.1.3.tar.gz
Algorithm Hash digest
SHA256 9954926bf75bf1fcf1a40983c655bee3990841d0ec83dbbcb619e2c17ce3fc11
MD5 71c4aa0fd0bb6b5fadfc822203839442
BLAKE2b-256 dcee3702c50ded7f25568d7247f8dccf81c6cf5c6a11f961933dce90b5913629

See more details on using hashes here.

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

Hashes for plotbbox-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 1a8d9de365cfe1829caa3e44dba887a79e0555b19d98485ba51c07ed0ef6827c
MD5 ca3ac420602554a8ebca66dbf8329ffc
BLAKE2b-256 7f31cff627ddd9658b8e8a3bdd3bdca9a95bf668fae44ca0d0073661b05f6bb2

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page