import glob
import os,sys
import shutil
import numpy as np
import cv2
import matplotlib.pyplot as plt
import os, random, shutil,cv2
# labelDir = 'F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/test/label/'
# imageDir = 'F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/yes/test/image/'
# labelDir1 = 'F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/test/label/'
# imageDir1 = 'F:/project/Breast/InBreast/INBreast/Unet/data/Inbreast/no/test/image/'
import time
import os
import math
import sys
import os,os.path,shutil
import numpy as np
import re
txtPath = 'F:/project/Breast/InBreast/INBreast/outimgpatch/allouttxtpatch/'
imagePath = 'F:/project/Breast/InBreast/INBreast/outimgpatch/allcalcification/'
labelPath = 'F:/project/Breast/InBreast/INBreast/outimgpatch/allcalcificationimglabel/'
noimagePath = 'F:/project/Breast/InBreast/INBreast/outimgpatch/allnocalcification/'
nolabelPath = 'F:/project/Breast/InBreast/INBreast/outimgpatch/allnocalcificationlabel/'
imagePath1 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/images/'
labelPath1 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/labels/'
noimagePath1 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/noimages/'
nolabelPath1 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/nolabels/'
imagePath2 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/others/images/'
labelPath2 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/others/labels/'
noimagePath2 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/others/noimages/'
nolabelPath2 = 'F:/project/Breast/InBreast/INBreast/outimgpatch/newcalcification/others/nolabels/'
txtType = 'txt'
txtLists = os.listdir(txtPath) #列出文件夹下所有的目录与文件
print(txtLists)
# Read the points(before 11:30,712)
# Convert points to digital form(before 2:30,712)
# Obtain the right batch(before 5.00, 712)
for filename in txtLists:
print(filename)
name = filename[:-4] + '.png'
print(name)
with open(txtPath + filename, 'r') as file:
lines = file.readlines()
dataset = [[] for i in range(len(lines))]
for i in range(len(dataset)):
dataset[i][:] = (item for item in lines[i].strip().split(',')) # 逐行读取数据
print(i)
if os.path.exists(imagePath + name):
if i > 3:
print("yes")
shutil.copy(imagePath + name, imagePath2 + name)
shutil.copy(labelPath + name, labelPath2 + name)
shutil.copy(noimagePath + name, noimagePath2 + name)
shutil.copy(nolabelPath + name, nolabelPath2 + name)
if i <= 3:
print("no")
shutil.copy(imagePath + name, imagePath1 + name)
shutil.copy(labelPath + name, labelPath1 + name)
shutil.copy(noimagePath + name, noimagePath1 + name)
shutil.copy(nolabelPath + name, nolabelPath1 + name)
# print("dateset:", dataset)
# # print(type(dataset[0][0]))
# # print(dataset.__sizeof__())
# u = np.array(dataset)
# for i in range(u.shape[0]):
# # print(u[i,0][0])
# findNumber = u[i,0].find(" ")
# # print(findNumber)
# x = round(float(u[i, 0][0:findNumber]))
# findNumber1 = u[i, 0][findNumber+1:].find(" ")
# y = round(float(u[i, 0][findNumber+1: findNumber + findNumber1]))
# print(x,y)
# 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)