为对象检测任务绘制漂亮边界框的包
项目描述
绘图盒
为对象检测任务绘制漂亮边界框的包
安装
pip install plotbbox
示例用法
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)
阅读docs/example.py并运行以获取更多详细信息！
下载文件
下载适用于您平台的文件。如果您不确定要选择哪个，请了解有关安装包的更多信息。
源分布
plotbbox-0.1.3.tar.gz
（151.5 kB
查看哈希）
内置分布
plotbbox-0.1.3-py3-none-any.whl
（151.1 kB
查看哈希）