【DL】解决 "No module named 'utils.nms.cpu_nms' "报错 'cpu_nms.py'

本文解决在运行开源模型时遇到的cpu-nms模块未找到错误,提供cpu_nms.pyx文件的替换方案,并建议在遇到gpu-nms问题时,可在utils/nms_wrapper.py中注释掉gpu相关代码,强制使用cpu版本。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

跑一些开源模型,会遇到这样的报错:

cpu-nms 报错

ModuleNotFoundError: No module named ‘utils.nms.cpu_nms’

解决方案:把 cpu_nms.pyx(一般路径为 ./utils/nms/cpu_nms.pyx) 替换为 cpu_nms.py.

cpu_nms.py 下载

这样, cpu-nms 就可以用了

gpu-nms 报错

懒得处理,可以直接在 ./utils/nms_wrapper.py注释掉 gpu 相关的,强制使用前面调好的 cpu

Ref

yolo_obb使用时报错,C:\Users\metrology2\PycharmProjects\yolo\venv\Scripts\python.exe C:/Users/metrology2/PycharmProjects/yolo/yolov5_obb-master/detect.py Traceback (most recent call last): File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\detect.py", line 30, in <module> from models.common import DetectMultiBackend File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\models\common.py", line 23, in <module> from utils.datasets import exif_transpose, letterbox File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\datasets.py", line 28, in <module> from utils.augmentations import Albumentations, augment_hsv, copy_paste, letterbox, mixup, random_perspective File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\augmentations.py", line 12, in <module> from utils.general import LOGGER, check_version, colorstr, resample_segments, segment2box File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\general.py", line 35, in <module> from utils.nms_rotated import obb_nms File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\nms_rotated\__init__.py", line 1, in <module> from .nms_rotated_wrapper import obb_nms, poly_nms File "C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\nms_rotated\nms_rotated_wrapper.py", line 4, in <module> from . import nms_rotated_ext ImportError: cannot import name 'nms_rotated_ext' from partially initialized module 'utils.nms_rotated' (most likely due to a circular import) (C:\Users\metrology2\PycharmProjects\yolo\yolov5_obb-master\utils\nms_rotated\__init__.py) Process finished with exit code 1
最新发布
07-08
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值