Open apanand14 opened 1 year ago
俺也一样
Seems your model's output dimension is not as expected. You need to check how you converted Torch model to Onnx. You can use ultralytics yolo to export to onnx.
command :
yolo export model=<torch model path> format=onnx batch=16|32 dynamic=True device=0
Use batch
, dynamic
as per your requirement.
Hello,
I would like to run inference using my converted onnx model but somehow I'm getting this error. Can you please look into it if possible? Thank you in advance