fix: adjusted path for directory input

frankkramer-lab · Jan 23, 2024 · 78c7b94 · 78c7b94
1 parent 8f5ad29
commit 78c7b94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/main.py b/code/main.py
@@ -112,7 +112,7 @@
 PREDICTION_PATH = args.prediction
 
 INPUTS = args.input
-if os.path.isdir(INPUTS[0]) : INPUTS = os.listdir(INPUTS[0])
+if os.path.isdir(INPUTS[0]) : INPUTS = [os.path.join(INPUTS[0], x) for x in os.listdir(INPUTS[0])]
 
 RES_PATH = args.output  # location of full slides
 if not os.path.exists(RES_PATH):