您当前的位置: 首页 > 
  • 1浏览

    0关注

    483博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

ModuleNotFoundError: No module named ‘lib.external.nms‘

高精度计算机视觉 发布时间:2020-07-20 19:11:29 ,浏览量:1

最近在尝试CenterFace.pytorch/ CenterNet等网络,报错,

ModuleNotFoundError: No module named 'lib.external.nms'

在linux下,一般的解决办法是

cd src/lib/external make

但在windows下,发现有一个setup.py,懒得去设置环境什么的,不过,这个setup.py中,name="coco",非常不喜欢,因为coco系统中已经有不少extension了,所以我改成了external,当然你可以改成任何名称,

import numpy
from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize

extensions = [
    Extension(
        "nms", 
        ["nms.pyx"],
        extra_compile_args=[] #["-Wno-cpp", "-Wno-unused-function"]
    )
]

setup(
    name="external",
    ext_modules=cythonize(extensions),
    include_dirs=[numpy.get_include()]
)

python setup.py install

报告说安装到了这里,

copying build\lib.win-amd64-3.7\nms.cp37-win_amd64.pyd -> d:\Anaconda3\envs\tch37\Lib\site-packages running install_egg_info Writing d:\Anaconda3\envs\tch37\Lib\site-packages\external-0.0.0-py3.7.egg-info

我仔细看了一下,其实还另外安装了一个文件在同一目录下,叫做

nms.cp37-win_amd64.pyd

这下好办了,直接把

#from external.nms import soft_nms

改成

from nms import soft_nms 

然后就可以运行了。

 

 

关注
打赏
1661664439
查看更多评论
立即登录/注册

微信扫码登录

0.0382s