首页 > 其他 > 详细

筛选图像对

时间:2020-03-26 14:19:11      阅读:59      评论:0      收藏:0      [点我收藏+]

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:])   

 

筛选图像对

原文:https://www.cnblogs.com/wjjcjj/p/12574063.html

踩
(0)
赞
(0)
   
举报
评论 一句话评论(0)
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!