import os
import cv2
import random
import shutil
import numpy as np
def get_data_crop(k):
# 获取所有病人序列名称与所有训练文件名称
name_list, file_list = [], []
for root, dirs, files in os.walk('./seg'):
for f in files:
if f.split('-')[0] not in name_list:
name_list.append(f.split('-')[0])
file_list.append(root + '/' + f)
random.shuffle(name_list)
#k折交叉验证
for i in range(8,10):
fold_size = len(name_list) // k
test_name = name_list[i*fold_size:(i+1)*fold_size]
train_file, test_file = [], []
for file in file_list:
if file.split('/')[-1].split('-')[0] in test_name:
try:
shutil.copy(file,'.\\K10\\test_'+str(i+1)+'\\M\\'+file.split('/')[-1])
shutil.copy(file.replace('seg','img'), '.\\K10\\test_' + str(i+1) + '\\I\\' + file.split('/')[-1])
except Exception:
os.makedirs('.\\K10\\test_'+str(i+1)+'\\M\\')
shutil.copy(file, '.\\K10\\test_' + str(i+1) +'\\M\\' +file.split('/')[-1])
os.makedirs('.\\K10\\test_' + str(i+1) + '\\I\\')
shutil.copy(file.replace('seg', 'img'),'.\\K10\\test_' + str(i+1) + '\\I\\' + file.split('/')[-1])
else:
try:
shutil.copy(file, '.\\K10\\train_' + str(i+1) +'\\M\\'+file.split('/')[-1])
shutil.copy(file.replace('seg', 'img'),'.\\K10\\train_' + str(i+1) + '\\I\\' + file.split('/')[-1])
except Exception:
print(file)
print(i+1)
os.makedirs('.\\K10\\train_'+str(i+1) +'\\M\\')
shutil.copy(file, '.\\K10\\train_'+str(i+1) + '\\M\\' +file.split('/')[-1])
os.makedirs('.\\K10\\train_' + str(i+1) + '\\I\\' )
shutil.copy(file.replace('seg', 'img'),'.\\K10\\train_' + str(i+1) + '\\I\\' + file.split('/')[-1])
get_data_crop(10)
10折交叉验证
最新推荐文章于 2023-12-18 09:40:52 发布