Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CANNOT CREATE 'convert_yolov3_coco.pth'. ~~No module named 'utils.nms.cpu_nms'~~ #7

Open
Hikaru-Furuta opened this issue Jan 15, 2021 · 0 comments

Comments

@Hikaru-Furuta
Copy link

Hikaru-Furuta commented Jan 15, 2021

I've been trying to create 'convert_yolov3_coco.pth' by following the README.
But the following error occurs.
Looking into the directory, utils.nms, there is cpu_nms.pyx instead cpu_nms.py.
How can I import DEFs from .pyx file??
Please teach me how to address this problem.

@user:~/git/YOLOv3_Pytorch$ python -m convert_darknet.py --version coco --weights ./weights/yolov3.weights --save_name ./weights/convert_yolov3_coco.pth
Traceback (most recent call last):
  File "/home/nel/anaconda3/envs/CenterNet_ONNX/lib/python3.6/runpy.py", line 183, in _run_module_as_main
    mod_name, mod_spec, code = _get_module_details(mod_name, _Error)
  File "/home/nel/anaconda3/envs/CenterNet_ONNX/lib/python3.6/runpy.py", line 109, in _get_module_details
    __import__(pkg_name)
  File "/home/nel/git/YOLOv3_Pytorch/convert_darknet.py", line 10, in <module>
    from model.yolo import Yolov3
  File "/home/nel/git/YOLOv3_Pytorch/model/yolo.py", line 11, in <module>
    from utils.box_utils import permute_sigmoid, decode
  File "/home/nel/git/YOLOv3_Pytorch/utils/box_utils.py", line 11, in <module>
    from utils.nms_wrapper import nms
  File "/home/nel/git/YOLOv3_Pytorch/utils/nms_wrapper.py", line 8, in <module>
    from .nms.cpu_nms import cpu_nms, cpu_soft_nms
ModuleNotFoundError: No module named 'utils.nms.cpu_nms'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant