python中的新特性。我想对一个文件夹中的不同图像应用算法，并将新图像保存到另一个文件夹中。生物学研究图像

Question

我想反转黑白图像中的颜色，然后用以下代码更改透明背景：

imgg = Image.open('HSPl4_E5_LP8.png')
data = np.array(imgg)

converted = np.where(data == 255, 0, 255)

imgg = Image.fromarray(converted.astype('uint8'))

imgg.save('new HSPl4_E5_LP8.png')

及

from PIL import Image
   

img = Image.open('new HSPl4_E5_LP8.png')
img = img.convert("RGBA")
datas = img.getdata()
     
       
newData = []
for item in datas:
    if item[0] == 255 and item[1] == 255 and item[2] == 255:
        newData.append((255, 255, 255, 0))#0 és la alfa de rgba i significa 0 opacity.
   
    else:
            newData.append(item)
            
    
img.putdata(newData)
img.save("HSPl4_E5_LP8 transparent.png", "PNG")

然后我想在一个文件夹中的几个图像中重复这个过程。然后，我想将新图像和更改保存在另一个文件夹中。但我没有找到一个办法让它发挥作用

Answer 1

假设apply_algo是一个函数，它为输入图像获取一个路径对象，并返回转换后的PIL.image对象，则可以使用pathlib实现此目的

from pathlib import Path


def process_files(source: str, dstn: str):
    dstn = Path(dstn)
    source = Path(source)
    # check if input strings are directories or not.
    if not (source.is_dir() and dstn.is_dir()):
        raise Exception("Source and Dstn must be directories")
    # use rglob if you want to pick files from subdirectories as well
    for path in source.glob("*"):
        if path.is_file():
            output_img = apply_algo(path)
            output_img.save(dstn / path.name(), "PNG")