Python脚本012：创建文件夹并把同id的图片放入

Author:Xuliu
Time:20211106
Function:create directories using the ids and put the same id into the directories named with id

Input：

第一个是id名

Output：

# 导入对路径操作的库
import sys
import os
from IPython import embed
import shutil
import glob

def mkDir(path,id):
    isExists = os.path.exists(os.path.join(path, id))
    if not isExists:						#判断如果文件不存在,则创建
        os.makedirs(os.path.join(path, id))


if __name__ == '__main__':
    # 读一下pic_path有多少文件
    # 获取id文件名是以id命名
    # os.makedirs(path+str(i))
    # 把同id的放到指定文件夹里
    cls_path = '/home/jy/xl/workstation/Datasets/Car/id_Car_dataset_Cut'  # 设置创建后文件夹存放的位置
    pic_path = '/home/jy/xl/workstation/Datasets/Car/Car_dataset_Cut'  # 图片存放的位置

    img_names = os.listdir(pic_path)

    for img_name in img_names:
        str_list = img_name.split("_")
        # str_list = img_path.split("\\")[-1].split("_")
        pid = str_list[0]
        cid = str_list[1][1:2]
        mkDir(cls_path, pid)
        cut_pic = os.path.join(pic_path, img_name)
        cls_pic = os.path.join(os.path.join(cls_path,  pid), img_name)
        # cls_pic = os.path.join(cls_path+pid, img_name)
        shutil.copy(cut_pic, cls_pic)
        # embed()