Невозможно прочитать имя файла из файла CSV и найти его в другой папке

Question

Мне нужно просмотреть записи файла CSV, который состоит из следующей структуры:

Img_File |Класс

---

img1.png |

img2.png |

...

Первый столбец содержит имена файлов, а другой столбец Класс пуст. Все эти файлы сохраняются в папке по пути PATH . Мне нужно

• пройти через каждое имя файла в CSV-файле
• получить доступ к файлу из местоположения PATH
• передать его в модель и сделать прогнозы (класс предсказания)
• сохранить прогнозируемый класс в другой столбец в строке обрабатываемого изображения.

Мне удалось сделать прогнозы для изображения, но я не могу сделатьто же самое для заданных файлов, упомянутых в файле CSV. Как мне действовать?

Вот ссылка на блокнот для воспроизведения проблемы: ССЫЛКА

PS: с моим кодом может быть более 1 проблемы.

Answer 1

Я дал ему немного времени и понял:

пройти через каждое имя файла в файле CSV

import pandas as pd dff = pd.read_csv(path, index_col="Column_with_all_filenames")

получить доступ к файлу из местоположения PATH

for img_name, Class in dff.iterrows(): if(".png" in img_name): #Check is the name is that of a PNG file if(find(img_name, path) ): #Check if the file was found in the path im=test_img_dir+'/'+img_name #Modify the path of the print(im) im = image.imread(im) #Make Predictions

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

      #Create a list of the [filename--Prediction]
      listof_rows.append([file_name, prediction])

      #After all the files have been processed...
      df=pd.DataFrame(listof_rows, columns=['Image_File', 'Class'])
      df.to_csv('Result1.csv', index=False)