报错ModuleNotFoundError: No module named ‘util.nms.cpu_nms’

本文介绍了在复现GitHub开源代码时遇到的ModuleNotFoundError问题,详细展示了如何在util.nms模块下添加cpu_nms.py文件以解决该错误。cpu_nms.py文件包含了实现CPU版非极大值抑制(Non Maximum Suppression, NMS)的函数,用于处理目标检测中的重叠框。代码中定义了max和min操作,并提供了nms函数来计算并去除重叠的检测框。
摘要由CSDN通过智能技术生成

在复现GitHub上的开源代码时,遇到报错ModuleNotFoundError: No module named ‘util.nms.cpu_nms’
这是文件编译失败,没有生成cpu_nms.py
在util.nms文件下添加cpu_nms.py文件,内容如下:

# cpu_nms.py
import numpy as np


def mymax(a, b):
    return a if a >= b else b


def mymin(a, b):
    return a if a <= b else b


def cpu_nms(dets, thresh):
    x1 = dets[:, 0]
    y1 = dets[:, 1]
    x2 = dets[:, 2]
    y2 = dets[:, 3]
    scores = dets[:, 4]
    areas = (x2 - x1 + 1) * (y2 - y1 + 1)
    order = scores.argsort()[::-1]
    ndets = dets.shape[0]
    suppressed = np.zeros((ndets), dtype=np.int)
    keep = []
    for _i in range(ndets):
        i = order[_i]
        if suppressed[i] == 1:
            continue
        keep.append(i)
        ix1 = x1[i]
        iy1 = y1[i]
        ix2 = x2[i]
        iy2 = y2[i]
        iarea = areas[i]
        for _j in range(_i + 1, ndets):
            j = order[_j]
            if suppressed[j] == 1:
                continue
            xx1 = mymax(ix1, x1[j])
            yy1 = mymax(iy1, y1[j])
            xx2 = mymin(ix2, x2[j])
            yy2 = mymin(iy2, y2[j])
            w = mymax(0.0, xx2 - xx1 + 1)
            h = mymax(0.0, yy2 - yy1 + 1)
            inter = w * h
            ovr = inter / (iarea + areas[j] - inter)
            if ovr >= thresh:
                suppressed[j] = 1
    return keep

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值