微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

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

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

使用 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 举报,一经查实,本站将立刻删除。