Как сделать так, чтобы каждое изображение имело соответствующий тег?

Question

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

      def get_augmt(image,model):
    i = 0
    img_list =[]
    for batch in model.flow (x= image,
                            batch_size = 1,
                            save_to_dir ='probando', 
                            save_prefix =" " + label,
                            save_format ='jpeg',
                            subset = None):
        i += 1
        img_list.append(batch)
        if i > 5:
            return img_list  ```


pr["img_aug"] = pr["images"].apply(lambda x: get_augmt(x,datagen))

Answer 1

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

Если в pr["image"] вы сможете прикрепить к каждому изображению метку, будет проще получить к ней доступ из функцииget_augmt.

В противном случае вы можете использоватьiвместо этого и код будет:

      def get_augmt(image,model):
    i = 0
    img_list =[]
    for batch in model.flow (x= image,
                            batch_size = 1,
                            save_to_dir ='probando', 
                            save_prefix =" " + str(i),
                            save_format ='jpeg',
                            subset = None):
        i += 1
        img_list.append(batch)
        if i > 5:
            return img_list  ```


pr["img_aug"] = pr["images"].apply(lambda x: get_augmt(x,datagen))