前言:
需要修改的文件夹A中的各类文件夹(a、b)中的图片。
---A
---a
--abcd1.jpg
--1abcd.jpg
---b
--b-1.jpg
--b-2.jpg
修改后的文件夹B的名称结构图:
---B
---a
--1.jpg
--2jpg
---b
--1.jpg
--2.jpg
代码实现:
import glob
import os
import threading
from PIL import Image
input_images_path = "./dataset"
output_images_path = "./data"
#将读取到的文件保存到指定文件夹中
def create_image(infile,index,dir):
os.path.splitext(infile)
im = Image.open(infile)
im.save(output_images_path + "/" + str(dir) + "/" + str(index) + ".JPEG", "JPEG") #存储路径
#读取文件夹中的全部图片
def start():
dirs = os.listdir(input_images_path )
for dir in dirs:
for index in range(1):
for infile in glob.glob(input_images_path + "/" + str(dir) + "/" + "images" + "/" + "*.JPEG"): # 数据来源
print(infile)
t = threading.Thread(target=create_image, args=(infile, index, dir))
t.start()
t.join()
index += 1
if __name__ == "__main__":
start()
[注]:这里需要提前在文件夹B中创建文件夹A中各类的空子文件夹。具体方法点击这里。