Странное поведение при классификации изображений - Искусственный интеллект - Ответ 13329889

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
 
def dist(x,y):
    return np.sqrt(np.sum((x-y)**2))
 
def printFigure(figure):
    plt.figure()
    plt.imshow(figure, cmap = plt.cm.gray_r, interpolation = 'nearest')
    plt.show()
    
def loadFiles(count, start, dotType):
    res = np.empty(count, dtype=object)
    for i in range(0, count):
        path = start + str(i+1) + dotType
        img = Image.open(path)
        res[i] = np.asarray(img, dtype='uint8')
    return res
 
circles = loadFiles(10, 'D:\\figure\\learn\\circle\\c', '.png')
rects = loadFiles(10, 'D:\\figure\\learn\\rect\\r', '.png')
triangles = loadFiles(10, 'D:\\figure\\learn\\triangle\\t', '.png')
 
labels = np.empty(30, dtype=object)
for i in range(0, 10):
    labels[i] = 'circle'
for i in range(10, 20):
    labels[i] = 'rect'
for i in range(20, 30):
    labels[i] = 'triangle'
 
figures = np.concatenate((circles, rects, triangles))
 
 
 
 
img = Image.open('D:\\figure\\learn\\co1.png')
arrControl = np.asarray(img, dtype='uint8')
 
printFigure(arrControl)
 
num = len(figures)
distance = np.zeros(num)
for i in range(num):
    distance[i] = dist(figures[i], arrControl)
    print( distance[i])
minIndex = np.argmin(distance)
 
print(labels[minIndex])
print(distance[minIndex])