我有一个包含300行和4列的IoU值的numpy数组。如果行中的每个元素都小于0.5,则选择行。我编写了一个尝试这样做的代码,但是它返回每个元素都为零的行。import numpy as npfor x in range(len(iou)): negative_boxes.append(iou[x])
如何选择每个元素都小于0.5的
我是ML的新手,我正在尝试实现自己的损失函数( IoU损失函数),但是得到了一个关于梯度的错误(“没有为任何变量提供梯度”)。因为损失应该最小化,所以我在函数的末尾放了1-IoU。这里是我的功能,更清楚的是,我不知道如何准确地解释我的问题: #roundig thefor each row (i.e. for each predicted y and a true y)
union = y_true.shape[1] # w