YOLOv3_Pytorch icon indicating copy to clipboard operation
YOLOv3_Pytorch copied to clipboard

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

Open Hikaru-Furuta opened this issue 4 years ago • 0 comments

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'

Hikaru-Furuta avatar Jan 15 '21 02:01 Hikaru-Furuta