检查边界框是否围绕最后一个右侧球

编程问答 2022-05-29

问题描述

假设我有一张包含绿色和红色圆圈的图像。

使用 yolo，它将生成包含边界框的图像。

我想检查绑定框是否检测到右侧的最后一个球。

在第一个示例中，边界框不是 x 坐标上的最后一个非白色像素。右侧还有其他球。

enter image description here

在第二张图片上，它是右侧 x 坐标上的最后一个非白色像素。

enter image description here

使用 numpy，我试图找到一种方法来完成它。可能获取绑定框像素的坐标，并检查它是否是最后一个非白色像素。

from PIL import Image
import numpy as np


img = Image.open('/your_pic.png')
pixel_mat = np.array(img.getdata())
width = img.size[0]

pixel_ind = np.where((pixel_mat[:,:3] > 0).any(axis=1))[0]
coordinate = np.concatenate(
    [
        (pixel_ind % width).reshape(-1,1),(pixel_ind // width).reshape(-1,],axis=1,)

你们知道如何完成它吗？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

computer-visionnumpyopencvopencvyolov5