首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

在detectron2中,如何在对象检测中仅在没有标签(类名)的图像上绘制边界?

在detectron2中,可以通过以下步骤在没有标签(类名)的图像上绘制边界:

  1. 首先,导入必要的库和模块:
代码语言:txt
复制
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg
from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog
  1. 加载预训练的模型配置文件和权重:
代码语言:txt
复制
cfg = get_cfg()
cfg.merge_from_file("path/to/config.yaml")
cfg.MODEL.WEIGHTS = "path/to/model_weights.pth"
  1. 创建一个预测器(predictor):
代码语言:txt
复制
predictor = DefaultPredictor(cfg)
  1. 加载要进行对象检测的图像:
代码语言:txt
复制
image = cv2.imread("path/to/image.jpg")
  1. 进行对象检测并获取预测结果:
代码语言:txt
复制
outputs = predictor(image)
  1. 获取边界框(bounding box)的坐标和类别标签:
代码语言:txt
复制
boxes = outputs["instances"].pred_boxes.tensor.cpu().numpy()
labels = outputs["instances"].pred_classes.cpu().numpy()
  1. 创建一个可视化器(visualizer)并绘制边界框:
代码语言:txt
复制
metadata = MetadataCatalog.get(cfg.DATASETS.TEST[0])
visualizer = Visualizer(image[:, :, ::-1], metadata=metadata, scale=1.0)
visualizer = visualizer.draw_box(boxes, edge_color='red', line_style='--')
  1. 获取绘制了边界框的图像:
代码语言:txt
复制
output_image = visualizer.get_image()[:, :, ::-1]
  1. 显示或保存绘制了边界框的图像:
代码语言:txt
复制
cv2.imshow("Object Detection", output_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

这样,就可以在没有标签(类名)的图像上绘制边界框了。

推荐的腾讯云相关产品:腾讯云AI计算平台(https://cloud.tencent.com/product/tcap)

相关搜索:如何使用tensorflow对象检测API统计检测到的对象(在边界框中)的数量如何使用tf.image.draw_bounding_boxes在原始图像上绘制边界框以显示检测到对象的位置?如何从绘制在图像上的线条中收集像素?在Tensorflow对象检测API中,如何计算多个边界框预测的IoU?在Android中,我如何在另一个图像上绘制图像,但仅在它们重叠的地方?获取TypeError:在使用struct.pack()时,'int‘类型的对象没有len(),但仅在类中在CImg中,如何在3D图像的特定切片上绘制文本?在Jupyter Notebook中绘制tune类的对象时出错,但在RStudio中工作-如何修复它?如何将图像保存或打印到在wxPanel上绘制的wxPython文件中在Java中,在外部类中创建对象时,如何简化多级嵌套静态类名的使用?如何在tkinter中销毁一个类在另一个类上的标签?如何使用grid.raster在列表中的每个ggplot对象上覆盖图像?如果我在组件的父div上创建了react ref,是否可以仅在该ref中通过类名获取元素?在SceneKit中,如何在保持绘制调用最小化的同时,在不同大小的对象上平铺纹理?在F#中如何在没有窗口的情况下绘制到屏幕上的像素?如何防止类中的类对象尝试在没有默认构造函数的情况下自动构造自身?在Python中,当类不在范围内时,对象上的函数如何仍然可用?如何使用Iphone上的Objective-C在控制器中初始化自定义类/对象pygame -如何在不使用类/对象、Sprite或任何其他模块的情况下按顺序显示存储在列表中的png图像?
相关搜索:
页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券