deftruth / yolov5face-toolkit Goto Github PK

Hi,
@DefTruth
do you have alignment and resize module for yolo5face ?

Best

does the onnxruntime include nms?

File "export.py", line 72, in
y = model(img) # dry run
File "E:\Anaconda\envs\yolo\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "F:\yolov5-face-master\models\yolo.py", line 222, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "F:\yolov5-face-master\models\yolo.py", line 238, in forward_once
x = m(x) # run
File "E:\Anaconda\envs\yolo\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "F:\yolov5-face-master\models\yolo.py", line 62, in forward
y = y + torch.cat((x[i][:, :, :, :, 0:5].sigmoid(),
IndexError: too many indices for tensor of dimension 4

这一步老是过不去？

output is type: float32[1,25200,16]

what does 16 mean here?

your code may have a readme?I couldn't find it