将文件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)