前言:
需要修改的文件夾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中各類的空子文件夾。具體方法點擊這裏。