若文件夹存在就创建文件夹否则删除重建

import os
import shutil
#若模型保存文件夹不存在，创建模型保存文件夹，若存在，删除重建
def mkr(path):
    if os.path.exists(path):
        shutil.rmtree(path)
        os.mkdir(path)
    else:
        os.mkdir(path)
if __name__ == '__main__':
    output_directory='model'
    mkr(output_directory)#文件夹先处理一下，避免人工删除文件夹