自动新建输出文件夹

2023-09-17 09:56:30 +08:00 · 2023-09-17 09:56:30 +08:00 · d5ba9b2cf4
commit d5ba9b2cf4
parent 2b45bd6d60
1 changed files with 1 additions and 0 deletions
--- a/ApneaDetection.py
+++ b/ApneaDetection.py
@ -60,6 +60,7 @@ def main(opt):
        result = SA_Detect(model_path, data, opt.batch_size)
        # result.to_csv(Path(opt.output) / (one_file.stem + "segment.csv"), index=False)
        if opt.output:
+            Path(opt.output).mkdir(parents=True, exist_ok=True)
            save_path = Path(opt.output) / (one_file.stem + ".csv")
        else:
            save_path = None