Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
Object Detection
# IPython Notebook code
import cv2
import matplotlib.pyplot as plt
%matplotlib inline
cctv_image = cv2.imread('0744_04.jpg')
# the 'cascade.xml' file is the file generated by the training script above
vehicle_classifier = cv2.CascadeClassifier('cascade.xml')
# various parameters can be passed to modify how objects are detected
vehicles = vehicle_classifier.detectMultiScale(cctv_image, 1.1, 2, maxSize=(100,100))
print 'Vehicles detected: %d' % (len(vehicles))
# draw a rectangle around every vehicle detected
for (x,y,w,h) in vehicles:
cv2.rectangle(cctv_image, (x,y), (x+w, y+h),(255,0,0),2)
plt.figure(figsize=(9,9))
plt.axis('off')
plt.imshow(cv2.cvtColor(cctv_image, cv2.COLOR_BGR2RGB))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment