소스 검색

Adding results directory if it doesn't already exist

Jason Antic 6 년 전
부모
커밋
57941117f8
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      fasterai/visualize.py

+ 1 - 0
fasterai/visualize.py

@@ -20,6 +20,7 @@ class ModelImageVisualizer():
     def __init__(self, filter:IFilter, results_dir:str=None):
         self.filter = filter
         self.results_dir=None if results_dir is None else Path(results_dir)
+        self.results_dir.mkdir(parents=True, exist_ok=True)
     
     def _clean_mem(self):
         return