Numpy : 关于np.finfo函数

用法在这个网页中讲的比较好,有例子

先上代码:

#这是一个faster-rcnn中bbox_tools中的代码
    eps = xp.finfo(height.dtype).eps
    height = xp.maximum(height, eps)
    width = xp.maximum(width, eps)

    dy = (base_ctr_y - ctr_y) / height
    dx = (base_ctr_x - ctr_x) / width
    dh = xp.log(base_height / height)
    dw = xp.log(base_width / width)

这里可以看到有除法,我们要考虑到除法的分母是不能为0的,而且式子中log内也不能为负数,不然会直接跳出显示错误。

代码中是用上半部分来处理的,eps开始的三行将可能出现的负数和零,使用eps来替换,这样就不会出现错误了。

finfo函数是根据height.dtype类型来获得信息,获得符合这个类型的float型,eps是取非负的最小值。

你可能感兴趣的:(python:numpy)