YOLO output filtering inserted.

sergiomsilva · Nov 5, 2018 · a4fda91 · a4fda91
1 parent 1ddeb71
commit a4fda91
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vehicle-detection.py b/vehicle-detection.py
@@ -41,6 +41,8 @@
 
  R = detect(vehicle_net, vehicle_meta, img_path ,thresh=vehicle_threshold)
 
+ R = [r for r in R if r[0] in ['car','bus']]
+
  print '\t\t%d cars found' % len(R)
 
  if len(R):