bbox-visualizer
这个包帮助用户在物体周围绘制边界框,无需进行繁琐的标签定位计算。它还提供了几种不同类型的可视化方法,可用于在识别物体后进行标注。
边界框坐标点的预期格式为:(xmin, ymin, xmax, ymax)
- 文档:https://bbox-visualizer.readthedocs.io
- 免费软件:MIT许可证
安装:
pip install bbox-visualizer
使用:
import bbox_visualizer as bbv
图片来自Unsplash上的Joshua Earle、Jonas Weckschmied和Sherzod Max。
图片 | 函数 |
---|---|
img = bbv.draw_rectangle(img, bbox) img = bbv.add_label(img, label, bbox, top=True) | |
img = bbv.draw_rectangle(img, bbox) img = bbv.add_T_label(img, label, bbox) | |
img = bbv.draw_flag_with_label(img, label, bbox) | |
img = bbv.draw_rectangle(img, bbox) img = bbv.add_label(img, label, bbox, top=False) | |
img = bbv.draw_rectangle(image, bbox, is_opaque=True) img = bbv.add_label(img, label, bbox, draw_bg=False, top=False) | |
img = bbv.draw_multiple_rectangles(img, bboxes) img = bbv.add_multiple_labels(img, labels, bboxes) | |
img = bbv.draw_multiple_flags_with_labels(img, labels, bboxes) | |
img = bbv.draw_multiple_rectangles(img, bboxes) img = bbv.add_multiple_T_labels(img, labels, bboxes) |
有一些可选的函数可以在同一张图像上绘制多个边界框和/或写入多个标签,但建议在循环中使用上述函数,以便完全控制可视化效果。
- bbv.draw_multiple_rectangles(img, bboxes)
- bbv.add_multiple_labels(img, labels, bboxes)
- bbv.add_multiple_T_labels(img, labels, bboxes)
- bbv.draw_multiple_flags_with_labels(img, labels, bboxes)
在上述示例中,bboxes
和labels
是列表。
致谢
本包使用Cookiecutter和audreyr/cookiecutter-pypackage
项目模板创建。