-
Notifications
You must be signed in to change notification settings - Fork 68
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
Cannot use yolov8-lite-s #3
Comments
Same issue here. I'm using Python 3.10 on Windows 11. I have this kind of error: results = face_detector.predict(img, verbose=False, show=True, conf=0.25)[0]
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\yolo\engine\model.py", line 252, in predict
return self.predictor.predict_cli(source=source) if is_cli else self.predictor(source=source, stream=stream)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\yolo\engine\predictor.py", line 157, in __call__
return list(self.stream_inference(source, model)) # merge list of Result into one
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\utils\_contextlib.py", line 35, in generator_context
response = gen.send(None)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\yolo\engine\predictor.py", line 221, in stream_inference
preds = self.model(im, augment=self.args.augment, visualize=visualize)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\nn\autobackend.py", line 313, in forward
y = self.model(im, augment=augment, visualize=visualize) if augment or visualize else self.model(im)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\nn\tasks.py", line 203, in forward
return self._forward_once(x, profile, visualize) # single-scale inference, train
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\nn\tasks.py", line 58, in _forward_once
x = m(x) # run
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\nn\modules.py", line 479, in forward
stem_1_out = self.stem_1(x)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\ultralytics\nn\modules.py", line 66, in forward_fuse
return self.act(self.conv(x))
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\conv.py", line 463, in forward
return self._conv_forward(input, self.weight, self.bias)
File "C:\Users\Vincent\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\conv.py", line 459, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
TypeError: conv2d() received an invalid combination of arguments - got (Tensor, Parameter, Parameter, tuple, tuple, tuple, int), but expected one of:
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, tuple of ints padding, tuple of ints dilation, int groups)
didn't match because some of the arguments have invalid types: (Tensor, Parameter, Parameter, tuple of (int, int), tuple of (int, int), tuple of (bool, bool), int)
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, str padding, tuple of ints dilation, int groups)
didn't match because some of the arguments have invalid types: (Tensor, Parameter, Parameter, tuple of (int, int), tuple of (int, int), tuple of (bool, bool), int) |
I also have this problem, have you solved it? |
Only Best, |
"ultralytics\nn\modules\block.py",line42
|
This is helpful ,but you need modify the yolov8-lite-s-pose.yaml and yolov8-lite-t-pose.yaml config file ; here the yaml `backbone: [from, number, module, args][ [ -1, 1, StemBlock, [32, 3, 2,None,1] ], # 0-P2/4 v5lite-e headhead:
]` lite-t-pose:::`backbone: [from, number, module, args][ [ -1, 1, StemBlock, [16, 3, 2,None,1] ], # 0-P2/4 v5lite-e headhead:
] |
I have created a fork repository clibdev/yolov8-face which contains yolov8-lite-t and yolov8-lite-s models compatibility fixes. |
Hi,
I can use
yolov8n
but cannot seem to be able to use the other checkpoints:yolov8-lite-s
andyolov8-lite-t
I tried installing latest version of
ultralytics
but then the weigths cannot be loaded.I also tried using the custom folder
ultralytics
from this repo. In that case the weights can be loaded but inference then does not work.The text was updated successfully, but these errors were encountered: