这几天在做一个数据集,由于不是很熟悉Linux下的命令,所以特地用了强大的python来做。我之前有一个数据集但是我只要里面名称带有composite和normals的图片,所以找了网上的文章看到了用shutil.copyfile来实现的方法。
# -*- coding: utf-8 -*-
import os,shutil
#shutil.copyfile应用,参数必须具体到文件名
def mycopyfile(srcfile,dstfile):
if not os.path.isfile(srcfile):
print("%s not exit!" % (srcfile))
else:
fpath,fname=os.path.split(dstfile)
if not os.path.exists(fpath):
os.makedirs(fpath)
shutil.copyfile(srcfile,dstfile)
#print("copy %s" % (srcfile,dstfile))
#这个是找到图片的实际地址
def find_filepath(dir1_name,dir2_name,i,file_name=''):
file_path=os.path.join(dir1_name ,dir2_name ,str(i)+file_name+'.png')
return file_path
#这个是我自己的实际应用
if __name__=='__main__':
dir_name=['airplane_test','bottle_test','bunny_test','car_test','motorbike_test','suzanne_test','teapot_test']
obj=['_composite','_normals']
cnt=0
for i in range(7):
direction=dir_name[i]
#print(direction)
for j in range(400):
print(direction,j)
source1=find_filepath('intrinsic',direction,j,obj[0]) #源地址
source2=find_filepath('intrinsic',direction,j,obj[1]) #源地址
dst1=find_filepath('image2norm','testA',cnt) #目标地址
dst2=find_filepath('image2norm','testB',cnt) #目标地址
mycopyfile(source1,dst1)
mycopyfile(source2,dst2)
cnt+=1
值得注意的是,shutil.copyfile(srcfile,dstfile)里面srcfile和dstfile必须是文件名,不能是文件夹。具体应用可以参考上面的main。
如果想要复制文件夹下的全部文件而又不清楚文件的具体数目,可以采用以下os.listdir的方法列出该文件夹下所有文件的名称,
file_path='./resuls'
dirs=os.listdir(file_path)
for dir in dirs:
print(dir)