cmdbug / TNN_Demo

🍉 移动端TNN部署学习笔记,支持Android与iOS。
GNU General Public License v3.0
71 stars 18 forks source link

您好,模型转为tnn 精度下降问题 #10

Open duoduo1018 opened 3 years ago

duoduo1018 commented 3 years ago

您好, 我用官方给的yolov5s.pt转为tnn,测试结果正常。 但把自己训练的pt转为tnn后,精度下降很多(跟pytorch版本对比),这是哪里的原因呢?

cmdbug commented 3 years ago

这个可能需要给官方提issues

ppogg commented 3 years ago

您好, 我用官方给的yolov5s.pt转为tnn,测试结果正常。 但把自己训练的pt转为tnn后,精度下降很多(跟pytorch版本对比),这是哪里的原因呢?

https://github.com/ppogg/shufflev2-yolov5

试一下这个把兄弟,我也想知道这个模型转TNN或者MNN会不会崩