筛选图像对

1、两个文件夹的图像合并到另一个文件夹。

复制代码
import os
import shutil
import numpy as np
Adir = './trainA/'
Bdir = './trainB/'
ABdir = './train0/'

num=0
for fi in os.listdir(Adir):
        shutil.copy(Adir+fi,ABdir+str(num)+'_'+fi)
        shutil.copy(Bdir+'0'+fi[1:],ABdir+str(num)+'_'+'0'+fi[1:])
        num = num + 1
        print (num)     
复制代码

2、筛选结束的图像分开到两个文件夹中。

复制代码
import os
import shutil
import numpy as np
Adir = './trainA/'
Bdir = './trainB/'
ABdir = './train0/'

for fi in os.listdir(ABdir):
        length = len(fi.spilt('_')[0])
        if fi.spilt('_')[1]=='0':
            shutil.copy(ABdir+fi,Bdir+fi[length+1:])
        else:
            shutil.copy(ABdir+fi,Adir+fi[length+1:])   
复制代码

 

posted @   皮卡皮卡妞  阅读(318)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示