python实现根据图片的名字批量转移图片到另一个文件夹

python实现根据图片的名字将图片批量从一个文件夹复制到另一个文件夹

import os
import shutil

#按名字选取图片从一个文件夹到另一个文件夹里
def getFileName(fileName):
    (filePath, tempFileName) = os.path.split(fileName)
    (shotname, extension) = os.path.splitext(tempFileName)
    return shotname,extension

source_dir ='E:/KYY/work/youdaofeng/train/'
label_dir = 'E:/KYY/电成像测井图像/分割样本/labels/60/'
annotion_dir = 'E:/KYY/work/youdaofeng/label'

img = os.listdir(source_dir)
s=[]
#遍历文件夹
for fileNum in img:
    if not os.path.isdir(fileNum):
        imgName = os.path.join(source_dir,fileNum) #判断是否是文件夹，不是文件夹才打开
        (imgPath, tempimagName) = os.path.split(imgName) #将路径与文件名分开
        (shotname,extension) = os.path.splitext(tempimagName) #将文件名文本和文件后缀分开

    #将取出来的文件名文本与特定后缀拼接，再与路径B拼接，得到B目录下的文件
    tempimagName='%s.png'%shotname
    imgName=os.path.join(label_dir,tempimagName)

    #根据得到的png文件名，将对应文件拷贝到指定目录C
    shutil.copy(imgName,annotion_dir)