Skip to content

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

Description

@Hikaru-Furuta

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'

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions