Как поменять код в пайтоне чтобы в конце тестировались не рандомные фото в нейронной сети а фото из папки?

Question

[Sultan555]

Это код из Kaggle и эта часть из тестирования Этот код предсказывает рандомное фото из набора данных
И у меня возникает вопрос как поменять код чтобы выдавалось не рандомное фото а фото из папки
вот ссылка на проект Kaggle https://www.kaggle.com/paoloripamonti/derma-diseas...
А этот код та самая часть тестирования нейронной сети

def load_image(filename):
    img = cv2.imread(os.path.join(BASE_DATASET_FOLDER, TEST_FOLDER, filename))
    img = cv2.resize(img, (IMAGE_SIZE[0], IMAGE_SIZE[1]) )
    img = img /255
    
    return img


def predict(image):
    probabilities = model.predict(np.asarray([img]))[0]
    class_idx = np.argmax(probabilities)
    
    return {classes[class_idx]: probabilities[class_idx]}

for idx, filename in enumerate(random.sample(test_generator.filenames, 10)):
    print("SOURCE: class: %s, file: %s" % (os.path.split(filename)[0], filename))
    
    img = load_image(filename)
    prediction = predict(img)
    print("PREDICTED: class: %s, confidence: %f" % (list(prediction.keys())[0], list(prediction.values())[0]))
    plt.imshow(img)
    plt.figure(idx)    
    plt.show()

Answer 1

[Владислав Премудрый]

Откуда filename?
Если ты кидаешь фото в эту папку то измени filename на имя файла