dog-qiuqiu / MobileNet-Yolo

MobileNetV2-YoloV3-Nano: 0.5BFlops 3MB HUAWEI P40: 6ms/img, YoloFace-500k:0.1Bflops 420KB:fire::fire::fire:
Other
1.72k stars 278 forks source link

mobilenetv2-yolov3-nano caffe与mnn模型运行不一致问题 #26

Open strivehub opened 4 years ago

strivehub commented 4 years ago

貌似yolov3-nano训练出的模型caffe输出正常,但是mnn输出就是乱的,我对比了两个模型每一层输出,发现是在layer72-upsample层开始不一样,感觉是

layer {

#    bottom: "layer71-route"
#    top: "layer72-upsample"
#    name: "layer72-upsample"
#    type: "Upsample"
#    upsample_param {
#        scale: 2
#    }
#}
layer {
    bottom: "layer71-route"
    top: "layer72-upsample"
    name: "layer72-upsample"
    type: "Interp"
    interp_param {
    height:20  #upsample h size
    width:20   #upsample w size
    }
}

模型转换上有点瑕疵,请问大大怎么修改才能解决?谢谢!!