MuhammadMoinFaisal/YOLOv8-DeepSORT-Object-Tracking

yolov8自己训练的模型,需要在ultralytics 8.0.3版本下训练才可以吗?

Chris-R-U opened this issue · 7 comments

File "predict.py", line 253, in predict
predictor()
File "/home/wr/miniconda3/envs/yolov8_py38/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
return func(*args, **kwargs)
File "/home/wr/deepsort_python/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/wr/deepsort_python/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/wr/deepsort_python/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/wr/deepsort_python/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/wr/miniconda3/envs/yolov8_py38/lib/python3.8/site-packages/torch/serialization.py", line 594, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "/home/wr/miniconda3/envs/yolov8_py38/lib/python3.8/site-packages/torch/serialization.py", line 853, in _load
result = unpickler.load()
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.

我更换了我自己训练的模型,在ultralytics 8.0.15环境下训练的yolov8s模型,但是替换后,报错上面

要在8.0.0上面訓練才可以動

Which platform are you running under ? linux or windows?

ubuntu18.04

要在8.0.0上面訓練才可以動

thanks ,我尝试一下

I have the same problem. Have you solved it?

该怎么做呢,必须在8.0.0上吗

8.0.0上面訓練才可以用,不過現在的yolov8有tracking的算法了,沒有一定要用deepsort的話可以直接用官方的tracking就行了