python:将文件从一个目录移动到另一个目录。附:nnUnet使用

在使用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,可以看这篇博客

posted @ 2022-04-23 18:48  九叶草  阅读(1217)  评论(0编辑  收藏  举报