背景:我使用之前训练的一个CPS分割模型,服务器上训练时精度有80%+,主要是分割人像、二分类。之后换了一个环境,在本地把pth转onnx,使用转的onnx文件进行推理发现效果很差,基本上啥都分割不出来。使用pth文件重新加载推理,效果一样,经过检查模型也完全一致,权重加载也没问题
到最后才发现,预处理步骤中少了归一化这一步
记录下自己的🐖b操作,决不再犯
如果你的模型输出结果十分异常,建议检查一下你是不是有和我一样🐖b操作
教训:要保证数据的预处理流程完全一致!!!!!!
背景:我使用之前训练的一个CPS分割模型,服务器上训练时精度有80%+,主要是分割人像、二分类。之后换了一个环境,在本地把pth转onnx,使用转的onnx文件进行推理发现效果很差,基本上啥都分割不出来。使用pth文件重新加载推理,效果一样,经过检查模型也完全一致,权重加载也没问题
到最后才发现,预处理步骤中少了归一化这一步
记录下自己的🐖b操作,决不再犯
如果你的模型输出结果十分异常,建议检查一下你是不是有和我一样🐖b操作
教训:要保证数据的预处理流程完全一致!!!!!!