Блог

Я пытаюсь добавить преобразования в набор данных видео, который у меня есть, и я уже извлек кадры. Я хочу добавить преобразования, но вместо сохранения в переменной в загрузчике данных я хотел бы экспортировать все преобразования в виде изображений в новый каталог, чтобы я мог использовать его для обучения модели с помощью видео — трансформатора Swin- https://github.com/SwinTransformer/Video-Swin-Transformer

Как я могу сохранить изображения?

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

Что-то вроде этого:

 transformed_images = []
for batch in dataloader:
    for video in batch:
        for frame in video:
            transformed_images.append(image)
 

В зависимости от того, где в ваших данных находится измерение последовательности