import glob import os,sys import shutil import numpy as np import cv2 import matplotlib.pyplot as plt # fileDir = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/allcalcification/‘ #源图片文件夹路径 # fileDir1 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/allcalcificationimglabel/‘ # fileDir2 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/allnocalcification/‘ # fileDir3 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/allnocalcificationlabel/‘ # tarDir = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/cal1/‘ # tarDir1 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/callabel1/‘ # tarDir2 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/calno1/‘ # tarDir3 = ‘F:/project/Breast/InBreast/INBreast/outimgpatch/calnolabel1/‘ import os, random, shutil,cv2 # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/test/image/‘ # fileDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/test/label/‘ # tarDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/test/image1/‘ # tarDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/test/label1/‘ # # if __name__ == ‘__main__‘: # txtLists = os.listdir(fileDir) # 列出文件夹下所有的目录与文件 # i = 0 # for filename in txtLists: # name = ‘0_‘+ str(i) + ‘_predict.png‘ # i = i + 1 # print(filename) # # # source = fileDir + filename # # # print(source) # shutil.copy(fileDir + filename, tarDir + name) # shutil.copy(fileDir1 + filename, tarDir1 + name) # # shutil.copy(fileDir2 + filename, tarDir2 + name) # # shutil.copy(fileDir3 + filename, tarDir3 + name) # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/train/image/‘ # fileDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/train/label/‘ # fileDir2 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/train/image/‘ # fileDir22 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/train/label/‘ # tarDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix/train/image/‘ # tarDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix/train/label/‘ # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/val/image/‘ # fileDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/val/label/‘ # fileDir2 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/val/image/‘ # fileDir22 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/val/label/‘ # tarDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix/val/image/‘ # tarDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix/val/label/‘ # # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/test/image/‘ # fileDir1 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/test/label/‘ # fileDir2 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/test/image/‘ # fileDir22 = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/test/label/‘ # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix2/test/image/‘ # fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/test/label/‘ fileDir = ‘F:/project/Breast/InBreast/INBreast/Unet/data/Inbreastmix2/test/image/‘ if __name__ == ‘__main__‘: txtLists = os.listdir(fileDir) # 列出文件夹下所有的目录与文件 print(txtLists) i = 0 for filename in txtLists: print(filename) # name = ‘0_‘+str(i)+‘_predict.png‘ # i =i + 2 # print(name) # shutil.copy(fileDir + filename, tarDir + name) # if filename.startswith(‘yes‘): # filename1 = filename[4:] # print(filename1) # filename2 = os.path.join(tarDir, filename1.split(‘.‘)[0] + ‘_yes.png‘) # print(filename2) # elif filename.startswith(‘no‘): # filename1 = filename[3:] # print(filename1) # filename2 = os.path.join(tarDir, filename1.split(‘.‘)[0] + ‘_no.png‘)#filename1 + ‘_no‘ # print(filename2) # # img = cv2.imread(fileDir + filename) # image = img[:,:,0] # cv2.imwrite(filename2,image) # name = ‘yes_‘ + filename#‘yes_0_‘+ str(i) + ‘_predict.png‘ # name1 = ‘no_‘ + filename#‘yes_0_‘+ str(i) + ‘_predict.png‘ # i = i + 1 # print(filename,name) # shutil.copy(fileDir + filename, tarDir + name) # shutil.copy(fileDir2 + filename, tarDir + name1) # shutil.copy(fileDir1 + filename, tarDir1 + name) # shutil.copy(fileDir22 + filename, tarDir1 + name1) # image1 = cv2.imread(fileDir + filename) # label1 = cv2.imread(fileDir1 + filename) # image2 = cv2.resize(image1,(448,448)) # label2 = cv2.resize(label1,(448,448)) # cv2.imwrite(fileDir + filename, image2) # cv2.imwrite(fileDir1 + filename,label2) # # source = fileDir + filename # # print(source) # shutil.copy(fileDir2 + filename, tarDir2 + name) # shutil.copy(fileDir3 + filename, tarDir3 + name)
原文:https://www.cnblogs.com/ziytong/p/11354565.html