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

ModuleNotFoundError: No module named 'ultralytics.nn.modules.conv'; 'ultralytics.nn.modules' is not a package #49

Open
Abhijeet241093 opened this issue Mar 18, 2024 · 5 comments

Comments

@Abhijeet241093
Copy link

When retrained your model yolov8l.pt with custom dataset and then used best. pt for prediction it giving following errors:
ModuleNotFoundError: No module named 'ultralytics.nn.modules.conv'; 'ultralytics.nn.modules' is not a package
!python predict.py model=yolov8l.pt source="output_video.mp4" show=True

[2024-03-18 20:54:18,878][root.tracker][INFO] - Loading weights from deep_sort_pytorch/deep_sort/deep/checkpoint/ckpt.t7... Done!
Ultralytics YOLOv8.0.3 🚀 Python-3.8.18 torch-2.2.1+cu121 CUDA:0 (NVIDIA GeForce RTX 3060, 12029MiB)
Error executing job with overrides: ['model=best.pt', 'source=output_video.mp4', 'show=True']
Traceback (most recent call last):
File "predict.py", line 326, in predict
predictor()
File "/home/tricubics/.local/lib/python3.8/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/home/tricubics/YOLOv8-DeepSORT-Object-Tracking/ultralytics/yolo/engine/predictor.py", line 164, in call
model = self.model if self.done_setup else self.setup(source, model)
File "/home/tricubics/YOLOv8-DeepSORT-Object-Tracking/ultralytics/yolo/engine/predictor.py", line 121, in setup
model = AutoBackend(model, device=device, dnn=self.args.dnn, fp16=self.args.half)
File "/home/tricubics/YOLOv8-DeepSORT-Object-Tracking/ultralytics/nn/autobackend.py", line 73, in init
model = attempt_load_weights(weights if isinstance(weights, list) else w,
File "/home/tricubics/YOLOv8-DeepSORT-Object-Tracking/ultralytics/nn/tasks.py", line 303, in attempt_load_weights
ckpt = torch.load(attempt_download(w), map_location='cpu') # load
File "/home/tricubics/.local/lib/python3.8/site-packages/torch/serialization.py", line 1026, in load
return _load(opened_zipfile,
File "/home/tricubics/.local/lib/python3.8/site-packages/torch/serialization.py", line 1438, in _load
result = unpickler.load()
File "/home/tricubics/.local/lib/python3.8/site-packages/torch/serialization.py", line 1431, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'ultralytics.nn.modules.conv'; 'ultralytics.nn.modules' is not a package

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

@reach-the-peak
Copy link

did you solve it?

@kz-jaha
Copy link

kz-jaha commented Mar 29, 2024

Same problem, has anyone solved it yet?

@WindyHu001
Copy link

same problem too.

@hjdasioh
Copy link

hjdasioh commented Apr 9, 2024

Did someone take it?

@FEIAFR11
Copy link

How can I solve it?

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

6 participants