我有100多个病人数据,每个病人又有10-40张不等的切片数量,所以我要在每个切片加入每个病人编号。
我要把所有病人数据放在一起。变成这样
import os
import os.path
import shutil
import glob
rootdir = "C:/Users/User/Desktop/UNet/"
new_root = "C:/Users/User\Desktop/UNet/train_label/"
for idx in range(107):
path = os.path.join(rootdir + str(idx+1) + "/mask/")
img_path = glob.glob(path+"/*.png")
last_img = img_path[-1]
print(idx+1)
if idx <= 8:
num = int(last_img[45:47])
# print(num)
elif idx>=99:
num = int(last_img[47:49])
else:
num = int(last_img[46:48])
# print(num)
# num = last_img[]
for idn in range(num):
if idn <= 8 :
img_path1 = os.path.join(path + "1000" + str(idn+1) + "_mask.png")
new_path = os.path.join(new_root + str(idx+1) + "_1000" + str(idn+1) + "_mask.png")
# print(new_path)
shutil.copyfile(img_path1,new_path)
else:
img_path1 = os.path.join(path + "100" + str(idn + 1) + "_mask.png")
new_path = os.path.join(new_root + str(idx + 1) + "_100" + str(idn + 1) + "_mask.png")
# print(new_path)
shutil.copyfile(img_path1, new_path)
print("finish__________________")