YOLOv3_Pytorch
YOLOv3_Pytorch copied to clipboard
CANNOT CREATE 'convert_yolov3_coco.pth'. ~~No module named 'utils.nms.cpu_nms'~~
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'