python 将图片按命名的类别保存到相应的文件夹中

Posted 2022-08-11 wuyi_all_in

将图片根据命名，分成7类，并保存到7个文件夹

---

import cv2
import os
i="0"
base_path="data/image/jaffe_48/"
imglist=os.listdir(base_path)
for img_path in imglist:
    #print(img_path)
    #print(img_path.find("AN")+1)
    if (img_path.find("AN")+1):
        i="0"
    elif (img_path.find("DI")+1):
        i = "1"
    elif (img_path.find("FE") + 1):
        i = "2"
    elif (img_path.find("HA")+1):
        i = "3"
    elif (img_path.find("SA") + 1):
        i = "4"
    elif (img_path.find("SU") + 1):
        i = "5"
    elif (img_path.find("NE") + 1):
        i = "6"
    if not os.path.exists(os.path.join(base_path,i)):
        os.makedirs(os.path.join(base_path,i))
    img=cv2.imread(os.path.join(base_path,img_path))
    f="/".format("data/image/jaffe_48",i)
    cv2.imwrite("/.jpg".format(f,img_path.split(".jpg")[0]),img)