想使用yolo-seg完成预测

[April-HAA]

想使用yolo-seg语义分割模型进行部署，该如何修改？我直接把模型换成seg模型，出现与get_out_tensor()相关的错误，找不到可以修改的地方。