image classification trained on stm32-h747 - results very biased

[fanronghua0123456]

I used stm32-h747 to train image classification, I used the image classification routine, but the actual effect is very biased, my quantized_model_confusion_matrix_validation_set is as shown below, is there something wrong with my configuration?

[Julian E.]

Hello @fanronghua0123456,

You mean that you trained a model with model zoo and deployed it on the stm32h747 disco?

In your case, I think you did provide images of only your label 1.
Your model predicted class 1 for almost all images, giving a good accuracy.

But you don't have any examples for the other labels, so you don't have metrics.

Please make sure to have a folder per class in your dataset folder if you are using model zoo.

Like:

datasets/

         MyDataset/

                         label1/

                                    img1.png

                                    img2.png

                                    ....

                         Label2/

                                    img1.png

                                    img2.png

                                    ....

Have a good day,

Julian