Как получить изображения из папки, используя python Открыть резюме в Google Colab, чтобы проверить производительность модели и распечатать все изображения в CSV-файл - PullRequest
Новая
       6

Как получить изображения из папки, используя python Открыть резюме в Google Colab, чтобы проверить производительность модели и распечатать все изображения в CSV-файл

1 голос
/ 04 марта 2020

Я обучил модель CNN в Google Colab, используя папку поезда. После построения модели я хочу проверить модель в тестовой папке (все изображения) и распечатать ее в CSV-файл. Я использовал код ниже, чтобы проверить модель на случайном изображении в тестовой папке. Но теперь я хочу проверить все изображения в тестовой папке и распечатать в CSV-файл.

Тестирование с использованием случайного изображения с кодом ниже 

 random_filename = random.choice([
    x for x in os.listdir(test_directory)
    if os.path.isfile(os.path.join(test_directory, x))])
 random_filename=test_directory + random_filename
 print(random_filename)
 img_array = cv2.imread(random_filename,cv2.IMREAD_COLOR)
 RGB_img = cv2.cvtColor(img_array, cv2.COLOR_BGR2RGB)
 new_img = cv2.resize(RGB_img,(img_size,img_size))
 print('Prediction : {}'.format(model.predict_classes(new_img.reshape(-1,img_size,img_size,3))))
 plt.imshow(new_img)
 plt.show()
...