Загрузка и маркировка пар фотографий

Question

Я создал ResNet18, чтобы определить, являются ли 2 человека братьями и сестрами или нет, предоставив изображение каждого из них (модель имеет input_size = 2). Мне нужно создать свой набор данных, в котором я буду указывать, какая пара является братьями и сестрами.

Я пытался:

training_set = train_datagen.flow_from_directory('training',
                                                 target_size=(28,28),
                                                 batch_size=32,
                                                 class_mode='binary')

И я получил training_set.classes array([0, 0, 0, 0, 1, 1, 1, 1])

для training_set.filenames

 'false\\false1\\_DSC5763.jpg',
 'false\\false2\\_DSC5751.jpg',
 'false\\false2\\_DSC5760.jpg',
 'siblings\\siblings1\\_DSC5751.jpg',
 'siblings\\siblings1\\_DSC5755_1.jpg',
 'siblings\\siblings2\\_DSC5760.jpg',
 'siblings\\siblings2\\_DSC5763.jpg'

Training_set.classes должно быть array([0, 0, 1, 1]), для моих целей. Как я могу это сделать?