use random image in demo.py

christian-rncl · Mar 29, 2018 · 2fb5631 · 2fb5631
1 parent 0f32903
commit 2fb5631
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/demo.py b/demo.py
@@ -68,8 +68,7 @@ class InferenceConfig(coco.CocoConfig):
 
 # Load a random image from the images folder
 file_names = next(os.walk(IMAGE_DIR))[2]
-#image = skimage.io.imread(os.path.join(IMAGE_DIR, random.choice(file_names)))
-image = skimage.io.imread(os.path.join(IMAGE_DIR, file_names[-1]))
+image = skimage.io.imread(os.path.join(IMAGE_DIR, random.choice(file_names)))
 
 # Run detection
 results = model.detect([image])