在使用nn-Unet做BraTS2019数据集预测时,预测文件分别生成了三类文件:.pkl .npz .nii.gz,我们需要的是.nii.gz文件。所以需要进行文件移动。
# coding:utf-8 import os,shutil import time src_path=f"C:\\Users\\wpx\\Desktop\\result\\nnunet\BraTS2019\OUTPUT_FOLDER\\" # 当前文件所在目录 target_path=f'C:\\Users\\wpx\\Desktop\\result\\nnunet\\BraTS2019\\submit\\' #移动的指定目录 while True: time.sleep(3) file_list=os.listdir(src_path) # 获取到当前目录下所有文件 if len(file_list)>0: # 如果目录不为空,那么进行移动 for file in file_list: file_end_name = file.split(".")[-1] if file_end_name == "gz": # 判断我们需要移动的文件,这里我需要移动的是.gz文件,其他文件不需要移动 shutil.move(src_path+file,target_path+file) # 利用这个方法进行移动 exit(0) # 移动完毕,进行退出
如何使用nn-Unet,可以看这篇博客。