把一个文件的图片转移到另一个文件

将文件path的图片转移到另一个文件内:

import os
import matplotlib.pyplot as plt
from torchvision import datasets
from PIL import Image
import torchvision.transforms as tran
from torchvision import utils

#collect是转移
def collect(path, i, filePath):
    for root, dirs, files in os.walk(path):
        for f in files:
            print(f)
            fl = path + '/' + f
            img = Image.open(fl)
            trans=tran.Compose([
                tran.CenterCrop(512),
                tran.ToTensor()
            ])
            img=trans(img)
            print(i)
            fn=os.path.join(filePath, str(i) + 'a.jpg')
            print(fn)
            utils.save_image(img, fn)
            i += 1
    return i


path = '被转目录'
i=1
filePath = '目标目录'
for dir in list(os.listdir(path)):
    i += 1
    cur_path=os.path.join(path,dir)
    print(cur_path)
    i=collect(cur_path, i, filePath)

你可能感兴趣的:(python)