From e2317bf0b821ffe3af011b5e33b58cfe81b48103 Mon Sep 17 00:00:00 2001
From: onb1259 <onb1259@onb.ac.at>
Date: Mon, 9 Oct 2023 16:45:44 +0200
Subject: [PATCH] fix not existing directories
---
extract_figures.ipynb | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/extract_figures.ipynb b/extract_figures.ipynb
index 79ca1cf..8a431bb 100644
--- a/extract_figures.ipynb
+++ b/extract_figures.ipynb
@@ -48,7 +48,8 @@
"IMAGES_DIR = 'images'\n",
"imgurls = get_imgurls_from_manifesturl(input_manifest)\n",
"imgpaths = create_paths_from_iiifurls(imgurls)\n",
- "shutil.rmtree(IMAGES_DIR)\n",
+ "if os.path.isdir(IMAGES_DIR):\n",
+ " shutil.rmtree(IMAGES_DIR)\n",
"os.mkdir(IMAGES_DIR)\n",
"imgpaths = [os.path.join(IMAGES_DIR, e) for e in imgpaths]\n",
"download_images_multithreded(imgurls, imgpaths, 10)"
@@ -66,7 +67,8 @@
"# Apply figure extraction model\n",
"\n",
"RESULTS_DIR = 'extracted_figures'\n",
- "shutil.rmtree(RESULTS_DIR)\n",
+ "if os.path.isdir(RESULTS_DIR):\n",
+ " shutil.rmtree(RESULTS_DIR)\n",
"os.mkdir(RESULTS_DIR)\n",
"model = YOLO('model_extract_figures.pt')\n",
"counter = 0\n",
--
GitLab