迭代文件夹打开，一张一张处理并保存图像

Question

我有一个文件夹 "Images-2"，它有 100 多个子文件夹，这些子文件夹每个文件夹包含一张图片。 def main() 打开每张图片，然后 def run(img) 获取图片并对其进行处理，但现在我无法将该图片保存在它的子文件夹中。

例如def main c:/Images-2/1/1.png（1是文件夹名称，所以我在Images-2中有100个文件夹）

if condition 会将处理后的图像 (zero.png) 保存在文件夹 Images-2/1/

中

如何处理 100 个文件夹，每个文件夹 1 张图片？

def run(img):
  data = img.load()
  width, height = img.size
  output_img = Image.new("RGB", (100, 100))
  Zero=np.zeros(shape=(100, 100),dtype=np.uint8)

  for (x, y) in labels:
            component = uf.find(labels[(x, y)])
            labels[(x, y)] = component
            path='C:/Python27/cclabel/Images-2/'
            if labels[(x, y)]==0:
                Zero[y][x]=int(255)
                Zeroth = Image.fromarray(Zero)
                for root, dirs in os.walk(path):
                    print root
                    print dirs
                    Zeroth.save(path+'Zero'+'.png','png')
def main():
    # Open the image
    path="C:/Python27/cclabel/Images-2/"
    for root, dirs, files in os.walk(path):
        for file_ in files:
            img = Image.open(os.path.join(root, file_))
            img = img.point(lambda p: p > 190 and 255)
            img = img.convert('1')
            (labels, output_img) = run(img)

if __name__ == "__main__": main()

Answer 1

您正在呼叫 os.walk 两次。那是你的问题。这就是我在评论中的意思：

def run(dirname, img):
    data = img.load()
    width, height = img.size
    output_img = Image.new("RGB", (100, 100))
    Zero=np.zeros(shape=(100, 100), dtype=np.uint8)

    for (x, y) in labels:
        component = uf.find(labels[(x, y)])
        labels[(x, y)] = component
        path = 'C:/Python27/cclabel/Images-2/'
        if labels[(x, y)] == 0:
            Zero[y][x] = 255
            Zeroth = Image.fromarray(Zero)
            Zeroth.save(os.path.join(dirname, 'Zero.png'), 'png')


def main():
    path = "C:/Python27/cclabel/Images-2/"
    for root, dirs, files in os.walk(path):
        for file_ in files:
            img = Image.open(os.path.join(root, file_))
            img = img.point(lambda p: p > 190 and 255)
            img = img.convert('1')
            (labels, output_img) = run(root, img)


if __name__ == "__main__":
    main()

迭代文件夹打开，一张一张处理并保存图像

Iterative folder open, process and save image one by one

python

python-imaging-library

subdirectory

os.walk

directory-traversal