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

最新推荐文章于 2024-11-08 12:13:02 发布

扑哧丶

最新推荐文章于 2024-11-08 12:13:02 发布

阅读量2.4k

点赞数 2

分类专栏：笔记文章标签： python

本文链接：https://blog.csdn.net/weixin_43873100/article/details/120473154

版权

笔记专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了在复现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