语义分割模型转 coreml 后 miou 精度下降明显

技术讨论 dzyjjpy ⋅ 于 3周前 ⋅ 78 阅读

做语义分割项目时,得到的模型从pytorch2onnx2coreml后,upsample层使用bilinear方式,转换得到的coreml模型精度下降非常厉害,miou从87%下降到82%。而通过同一个onnx模型,转换得到得m'n'n模型则精度几乎没有下降。所以应该不是onnx的锅。可能跟coreml转换后,upsample的层实现相关。
做过尝试过换成nearest mode,得到的模型精度不及bilinear方式。
欢迎有经验的小伙伴一起交流讨论 :)

成为第一个点赞的人吧 :bowtie:
回复数量: 0
暂无回复~
您需要登陆以后才能留下评论!