Блог

Итак, я создаю простую сверточную нейронную сеть для распознавания изображений, и у меня есть три папки, каждая из которых содержит изображения МРТ-сканирования другого типа опухоли головного мозга, и четвертая папка, содержащая изображения сканирования здорового мозга. Я пытаюсь визуализировать четыре изображения из каждой папки, и чтобы выбранные изображения были случайными. Когда я пытаюсь это сделать, отображаются только четыре изображения из первой папки.

Четыре папки помечены как глиома, менингиома, опухоль и гипофиз. Все они хранятся в учебном каталоге.

 #Training Directory 
train = "/content/Training/"

#Testing Directory
test = "/content/Testing/"

#Prediction Directory
predict = "/content/Predict/"

classes = os.listdir("/content/Training")

files_dict = {}

for c in classes: 
  files_dict[c] = list(map(lambda x :train c "/" x, os.listdir(train c)))

plt.figure(figsize=(20,20)) #Figure plot size
index = 0

for c in classes: 
  random.shuffle(files_dict[c])
  path_list = files_dict[c][:5]

for i in range(1, 5):
  index  = 1 
  plt.subplot(4, 4, index)
  plt.imshow(load_img(path_list[i]))
  plt.title(c)
  plt.grid(b=None)