将在一个文件夹里的人名图片移动到各自的文件夹里及其反操作

时间:2021-12-17 05:49:22

一个文件夹里都是图片,将属于同一个人的图片放到同一个文件夹

import os
import sys
import shutil

"""
将filePath文件下的图片保存在newFilePath文件夹下的相应子文件夹中
pic 是字典,存放每个图片要移到的子文件夹名
"""
#将在一个文件夹里的人名图片移动到各自的文件夹里
def moveImg(filePath):
    files = []
    dirs = os.listdir(filePath)
    for name in dirs:
        names = filePath + '/' + name

        if not os.path.isdir(names):
            files.append(str(name))
    for f in files:
        fl = filePath + '/' + f     #图片的绝对路径
        fname = str(f).split('_')[0]    #图片代表的人名
        if fname in dirs:   #已经有了人的文件夹

            path = filePath + '/' + fname
            shutil.move(fl,path)  # 移动文件或目录
        else:   #新人
            path = filePath + '/' + fname
            isExists = os.path.exists(path)
            if not isExists:
                os.makedirs(path)
            shutil.move(fl, path)  # 移动文件或目录


'''
        每次遍历的对象都是返回的是一个三元组(root,dirs,files)
            root 所指的是当前正在遍历的这个文件夹的本身的地址
            dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录)
            files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)
        '''
# moveImg()的反向操作
def removeImg(filePath):
    d = []
    dirs = os.listdir(filePath)

    for name in dirs:
        names = filePath + '/' + name

        if os.path.isdir(names):
            d.append(str(name))
    for i in range(len(d)): #对于每个人脸文件夹
        dirpath = filePath + '/' + str(d[i])
        files = os.listdir(dirpath)
        for f in files:
            path = dirpath + '/' + str(f)   #图片绝对路径
            shutil.move(path, filePath)  # 移动文件或目录

    for name in dirs:
        names = filePath + '/' + name

        if os.path.isdir(names):
            os.removedirs(names)


if __name__ == "__main__":
    #moveImg('/home/double/work/mydarknet/face')
    removeImg('/home/double/work/mydarknet/face')