Привет всем: при условии, что у меня есть 300 подпапок, каждая из которых содержит 15 ~ 19 изображений, расположенных в последовательном порядке, каждая серия изображений указывает событие, которое обозначено 0 или 1. Как объединить изображения каждой серии перед тренировкой?
Моя стратегия на данный момент:
from glob import glob
# root_folder contains the 300 subfolders
root_folder = r"D:\shooting_videos\example\Output"
# get the names of subfolders
subfolders = [f.path for f in os.scandir(root_folder) if f.is_dir()]
# loop over each subfolder, use glob to find the images in them
for i in subfolders:
img_data = glob (i + "\*.jpg")
print (img_data)
Хотя этот метод получает все изображения для меня, я надеялся, что формат будет таким: изображения в подпапке 1, изображения в подпапке 2, изображенияв подпапке 3 ... и т. д.
Тогда я могу использовать его в качестве X_train и создать Y_train с меткой для каждой серии, например:
X_train = [images in subfolder 1; images in subfolder 2; images in subfolder 3]
Y_train = [0, 1, 0]
Есть идеи? Заранее спасибо!