使用原始文件名保存图像

时间:2018-01-09 20:56:15

标签: python numpy python-imaging-library os.walk

我想用原始文件名保存图像,例如原始文件名是REGEXP_EXTRACT,因此处理后的图像名称应为hero
 我不知道如何在hero_Zero.png中传递文件名 下面的代码是正确的和修改的。

def run(dirs, img):

1 个答案:

答案 0 :(得分:0)

为函数定义添加一个额外的参数。不仅仅是获取目录和图像,还传递文件名。结果将如下所示:

def run(dirs, img, f_name):
  for (x, y) in labels:
        component = uf.find(labels[(x, y)])
        labels[(x, y)] = component   
        if labels[(x, y)]==0:
            Zero[y][x]=int(255)
            count=count+1
            if count<=43:
                continue
            elif count>43:
                Zeroth = Image.fromarray(Zero)
                Zeroth.save(os.path.join(dirs, f_name + '_Zero.png'), 'png')