KeyError Нет Индекс некоторых файлов изображений. проблема в SVM, диабетическая ретинопатия классификации с использованием SVM - PullRequest
0 голосов
/ 08 апреля 2019

Я проверил базу данных и CSV-файлы, и в сообщении об ошибке есть записи для указанного файла.Я также попытался удалить изображения ошибок, но не работал даже тогда.я нуб к питону.

path = "../train_resized/32x32/"

Files = []
Xlist = []
labels =[]
names = []

imlist = os.listdir(path)  
#print imlist[0]
imlist = imlist[1:]
im = np.array(Image.open(imlist[0]) #open one image to get the size
m,n = im.shape[0:2]


df_label = pd.read_csv('trainLabels.csv' )

def get_labels(i):
l = i.split(os.path.sep)[-1].split(".")[0]
st1 =  df_label.loc[[l]]
to_int = int(st1.values)
return to_int



def imagevector_label():
for file in os.listdir(path):
    Files.append(file)
File = Files[1:]
immatrix = np.array([np.array(Image.open(path+'/'+file)).flatten() for file in File],'f')    
for file in File:   
    names.append(file)
    labels.append(get_labels(file))

return immatrix, np.array(labels)        



img_vect, lab = imagevector_label()

from sklearn.model_selection import train_test_split


(trainRI, testRI, trainRL, testRL) = train_test_split(
img_vect, labels, test_size=0.25, random_state=42)

это сообщение об ошибке

enter image description here

Я ожидал, что изображения будут успешно загружены и SVM для успешной классификации.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...