如何使每个图像都有相应的标签

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))