ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
50.75k stars 16.34k forks source link

ModuleNotFoundError: No module named 'models.yolo' #6326

Closed Taitai6521 closed 2 years ago

Taitai6521 commented 2 years ago

Search before asking

YOLOv5 Component

No response

Bug

Namespace(agnostic_nms=False, augment=False, cfg='/content/drive/MyDrive/神様/Augmentedstartups/yolor/cfg/yolor_w6.cfg', classes=None, conf_thres=0.25, device='0', img_size=1280, iou_thres=0.5, names='data/coco.names', output='/content/drive/MyDrive/Dev/Soccer/Output', save_txt=False, source='/content/drive/MyDrive/神様/Augmentedstartups/yolor/inference/images/horses.jpg', update=False, view_img=False, weights=['/content/drive/MyDrive/神様/Augmentedstartups/yolor/yolor-w6.pt']) Traceback (most recent call last): File "detect.py", line 186, in detect() File "detect.py", line 44, in detect model.load_state_dict(torch.load(weights[0], map_location=device)['model']) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 607, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 882, in _load result = unpickler.load() File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 875, in find_class return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'models.yolo'

Environment

YoloR OS googlecolab

Minimal Reproducible Example

https://github.com/Taitai6521/ModuleNotFoundError-No-module-named-models.yolo- If you refer to the code here, you will be able to understand the contents. Thank you very much for your cooperation.

Additional

No response

Are you willing to submit a PR?

glenn-jocher commented 2 years ago

@Taitai6521 hi there! It looks like you've posted this issue in the wrong place. This is the YOLOv5 repo.

You should post YOLOR issues directly to the YOLOR repository.

Taitai6521 commented 2 years ago

hi, @glenn-jocher Sorry!!

Taitai6521 commented 2 years ago

P.S. https://github.com/WongKinYiu/yolor/issues/176

github-actions[bot] commented 2 years ago

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

Access additional Ultralytics ⚡ resources:

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!