From 214a4bd10ca57d77bdadd9ef49c914ba27aa94cf Mon Sep 17 00:00:00 2001 From: Till Raab Date: Thu, 26 Oct 2023 12:04:35 +0200 Subject: [PATCH] config rework --- confic.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/confic.py b/confic.py index b3c93de..f72d318 100644 --- a/confic.py +++ b/confic.py @@ -11,16 +11,14 @@ IMG_DPI = 256 DEVICE = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu') -TRAIN_DIR = 'data/train' - CLASSES = ['__backgroud__', '1'] NUM_CLASSES = len(CLASSES) +TRAIN_DIR = 'data/train' OUTDIR = 'model_outputs' -if not pathlib.Path(OUTDIR).exists(): - pathlib.Path(OUTDIR).mkdir(parents=True, exist_ok=True) - INFERENCE_OUTDIR = 'inference_outputs' -if not pathlib.Path(INFERENCE_OUTDIR).exists(): - pathlib.Path(INFERENCE_OUTDIR).mkdir(parents=True, exist_ok=True) \ No newline at end of file + +for required_folders in [TRAIN_DIR, OUTDIR, INFERENCE_OUTDIR]: + if not pathlib.Path(required_folders).exists(): + pathlib.Path(required_folders).mkdir(parents=True, exist_ok=True)