AI-Model-Zoo
AI-Model-Zoo copied to clipboard
Quantization of tf_mobilenetv1_1.0_imagenet_224_224_1.14G_1.1 with the DNNDK v3.1 package
Hello: I'm trying to port the tf_mobilenetv1_1.0 model to a ZedBoard. I'm using the DNNDK v3.1 package (Zynq-7000 family chips are not supported by Vitis-AI), and I'm having trouble with the quantization step. After quantization, there is a high decrease in accuracy of the model:
Accuracy | Frozen Graph | Quantized Graph |
---|---|---|
Top1 | 0.642000013589859 | 0.01399999968707561 |
Top5 | 0.851999986171723 | 0.06199999935925007 |
I believe the problem is that I'm using the wrong image pre-processing when calibrating (resize to 224x224, bgr2rgb and normalize between 0-1).
What image pre-processing has been used when training the tf_mobilenetv1_1.0 model?