diff --git a/ApneaDetection.py b/ApneaDetection.py
index f763433..7761f82 100644
--- 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