gpt4 book ai didi

python - 计算numpy数组的边界框

转载 作者:太空狗 更新时间:2023-10-30 00:36:50 26 4
gpt4 key购买 nike

我正在为一个简单的问题而苦苦挣扎。我有一个 numpy 形式的数组:

[[[ 1152.07507324   430.84799194]
[ 4107.82910156 413.95199585]
[ 4127.64941406 2872.32006836]
[ 1191.71643066 2906.11206055]]]

我想计算边界框,也就是说,我想要最左边、最上面、最右边和最下面的点。

这应该是一个正确的解决方案

[[[ 1152.07507324   413.95199585]
[ 4127.64941406 413.95199585]
[ 4127.64941406 2906.11206055]
[ 1152.07507324 2906.11206055]]]

我开发了一个讨厌的函数来解决这个问题,但我对它非常不满意,因为它不是真正的 pythonic/numpyic

def bounding_box(iterable):
minimum_x = min(iterable[0], key=lambda x:x[0])[0]
maximum_x = max(iterable[0], key=lambda x:x[0])[0]
minimum_y = min(iterable[0], key=lambda x:x[1])[1]
maximum_y = max(iterable[0], key=lambda x:x[1])[1]

return numpy.array([[(minimum_x, minimum_y), (maximum_x, minimum_y), (maximum_x, maximum_y), (minimum_x, maximum_y)]], dtype=numpy.float32)

您是否知道如何优化上述函数,也许使用 numpy 内置函数?

最佳答案

使用 numpy.minnumpy.max 内置函数:

def bounding_box(iterable):
min_x, min_y = numpy.min(iterable[0], axis=0)
max_x, max_y = numpy.max(iterable[0], axis=0)
return numpy.array([(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)])

关于python - 计算numpy数组的边界框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/12443688/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com