Klasifikace ručně psaného čísla

Model v onnx formátu byl natrénován na MNIST datasetu s přesností 99.29 %.

mnist_net = importONNXNetwork('MNIST_99.29.onnx','OutputLayerType','classification','ClassNames',{'0','1','2','3','4','5','6','7','8','9'}); %improt sítě

A = rgb2gray(imread('single_handwritten_digit.png'));%Načíst obrázek a převést na šedotón
A_resized = imresize(A,[28 28]);%Zmenšit na [28x28] cožodpovídá vstupní vrstvě neuronové sítě

[label, score] = classify(mnist_net,A_resized);
A_withText = insertText(A,[1 1], cellstr("classified as " +  string(label)+newline+ "("+ num2str(max(score)*100,'%0.2f %%)')) ,'FontSize',28); % přidat text s výsledkem kalsifikace
imshow(A_withText)

• Author:
• tesar-tech