Создание загрузчика / генератора данных изображения - PullRequest
Новая
       14

Создание загрузчика / генератора данных изображения

0 голосов
/ 28 мая 2020

Я добавил несколько изображений (png-файлы) и метки (csv-файл) для этих изображений на свой Google Диск. Подпапки (s01_l01, s01_l02, s02_l01 et c.), Которые содержат изображения и csv-файл с метками, хранятся в материнском каталоге 2017-IWT4S-CarsReId_LP-dataset. Это путь к материнскому каталогу: '/content/gdrive/My Drive/data/2017-IWT4S-CarsReId_LP-dataset/*'

Я прочитал в именах изображений, используя этот код: 

folders = glob.glob('/content/gdrive/My Drive/data/2017-IWT4S-CarsReId_LP-dataset/*')
imagenames_list = []
for folder in folders:
    for f in glob.glob(folder+'/*.png'):
        imagenames_list.append(f)

imagenames_list[0]
'/content/gdrive/My Drive/data/2017-IWT4S-CarsReId_LP-dataset/s01_l01/1_1.png'

Я прочитал в CSV-файле, используя этот код: 

track_id = []
image_path = []
lp = []
train = []

with open('/content/gdrive/My Drive/data/2017-IWT4S-CarsReId_LP-dataset/trainVal.csv') as csvDataFile:
    csvReader = csv.reader(csvDataFile)
    for row in csvReader:
        track_id.append(row[0])
        image_path.append(row[1])
        lp.append(row[2])
        train.append(row[3])

Список image_path выглядит так: 

  image_path[0]
  's01_l01/1_1.png'

Список lp выглядит так: 

lp[0]
'7C24698'

Из этих двух списков я создал список «образцы» который содержит image_path и lp, например: 

samples[:5]
[['s01_l01/1_1.png', '7C24698'],
 ['s01_l01/2_1.png', '7C24698'],
 ['s01_l01/2_2.png', '7C24698'],
 ['s01_l01/2_3.png', '7C24698'],
 ['s01_l01/2_4.png', '7C24698']]

Как мне теперь приступить к созданию загрузчика или генератора данных изображения? Я хотел бы создать один, потому что у меня много изображений и я не могу загрузить все сразу ... Вместо этого я хотел бы «загружать их партиями».

Большое спасибо!

...