Image folder argument was not used

mvcaro · Jun 4, 2018 · dd57170 · dd57170
1 parent bc4dfbc
commit dd57170
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -45,7 +45,7 @@
 
 model.eval() # Set in evaluation mode
 
-dataloader = DataLoader(ImageFolder('data/samples', img_size=opt.img_size),
+dataloader = DataLoader(ImageFolder(opt.image_folder, img_size=opt.img_size),
                         batch_size=opt.batch_size, shuffle=False, num_workers=opt.n_cpu)
 
 classes = load_classes(opt.class_path) # Extracts class labels from file