From 511fbd393638e822679bd2bc053ceafa4c1241fb Mon Sep 17 00:00:00 2001 From: rosaliabq Date: Tue, 9 Jan 2018 16:23:24 +0000 Subject: [PATCH] fixing input issues --- demo.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/demo.py b/demo.py index 11312f01ad..efdd464d39 100755 --- a/demo.py +++ b/demo.py @@ -114,11 +114,14 @@ def main(): IMAGETest_DIR = "/home/stephen/Videos/mightyAI/valeo_imgs" print("Preset input directory", IMAGETest_DIR) file_names = next(os.walk(IMAGETest_DIR))[2] + if not dirs.get_inputDir().endswith(".txt") and dirs.get_inputDir()!="": + print("Input directory", IMAGETest_DIR) + file_names = next(os.walk(IMAGETest_DIR))[2] if dirs.get_outputDir()=="": #IMAGETestResult_DIR = "/home/stephen/Videos/France_TestVideo/PBCS/results" IMAGETestResult_DIR = "/home/stephen/Videos/mightyAI/testResults" print("Preset output directory", IMAGETestResult_DIR) - file_names = next(os.walk(IMAGETest_DIR))[2] + file_names = next(os.walk(IMAGETestResult_DIR))[2] # Load a random image from the images folder