From 49790bc8b3e7adc9aaf7167093521999994d43b5 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Mon, 27 Apr 2020 21:06:22 +0200
Subject: [PATCH] [engine.predictor] Do not report directory creation
---
bob/ip/binseg/engine/predictor.py | 18 ++++--------------
1 file changed, 4 insertions(+), 14 deletions(-)
diff --git a/bob/ip/binseg/engine/predictor.py b/bob/ip/binseg/engine/predictor.py
index f4fe00d9..4e4640f0 100644
--- a/bob/ip/binseg/engine/predictor.py
+++ b/bob/ip/binseg/engine/predictor.py
@@ -40,10 +40,7 @@ def _save_hdf5(stem, prob, output_folder):
fullpath = os.path.join(output_folder, f"{stem}.hdf5")
tqdm.write(f"Saving {fullpath}...")
- fulldir = os.path.dirname(fullpath)
- if not os.path.exists(fulldir):
- tqdm.write(f"Creating directory {fulldir}...")
- os.makedirs(fulldir, exist_ok=True)
+ os.makedirs(os.path.dirname(fullpath), exist_ok=True)
with h5py.File(fullpath, 'w') as f:
data = prob.cpu().squeeze(0).numpy()
f.create_dataset("array", data=data, compression="gzip",
@@ -71,10 +68,7 @@ def _save_image(stem, extension, data, output_folder):
fullpath = os.path.join(output_folder, stem + extension)
tqdm.write(f"Saving {fullpath}...")
- fulldir = os.path.dirname(fullpath)
- if not os.path.exists(fulldir):
- tqdm.write(f"Creating directory {fulldir}...")
- os.makedirs(fulldir, exist_ok=True)
+ os.makedirs(os.path.dirname(fullpath), exist_ok=True)
data.save(fullpath)
@@ -128,13 +122,9 @@ def run(model, data_loader, device, output_folder, overlayed_folder):
"""
logger.info(f"Output folder: {output_folder}")
- logger.info(f"Device: {device}")
-
- if not os.path.exists(output_folder):
- logger.debug(f"Creating output directory '{output_folder}'...")
- # protect against concurrent access - exist_ok=True
- os.makedirs(output_folder, exist_ok=True)
+ os.makedirs(output_folder, exist_ok=True)
+ logger.info(f"Device: {device}")
model.eval().to(device)
# Sigmoid for predictions
sigmoid = torch.nn.Sigmoid()
--
GitLab