记录yolo v8模型训练后使用.pt模型推理正常 改用转换后的.onnx模型推理出现WARNING ⚠️ Unable to automatically guess model task, assu

yolo v8版本:‘8.0.228’

出现错误:

WARNING ⚠️ Unable to automatically guess model task, assuming 'task=detect'. Explicitly define task for your model, i.e. 'task=detect', 'segment', 'classify', or 'pose'.

解决办法:

将训练出的模型名字best.pt修改为best-seg.pt然后在转成onnx模型就可以了,个人猜可能源码是根据模型名字判断模型任务的

你可能感兴趣的:(问题相关,YOLO,深度学习,人工智能)