1+ #importing open computer vision!
2+ import cv2
3+
4+ #Test video
5+ img_file = './assets/car1.avi'
6+
7+ #trained detection algorithm
8+ bus_classifier = cv2 .CascadeClassifier ('bus.xml' )
9+ car_classifier = cv2 .CascadeClassifier ('cars.xml' )
10+ pedestrian_classifier = cv2 .CascadeClassifier ('pedestrian.xml' )
11+ bike_classifier = cv2 .CascadeClassifier ('bike.xml' )
12+
13+ #giving the image to openCV
14+ vid = cv2 .VideoCapture (img_file )
15+
16+ #run until the car turns off or crashes and burns to ashes
17+ while True :
18+ #gives a touple
19+ (frame_read ,frame ) = vid .read ()
20+
21+ if frame_read :
22+ #convert the image to garyscale
23+ gimg = cv2 .cvtColor (frame ,cv2 .COLOR_BGR2GRAY )
24+
25+ #Detect the everything with their respective algorithms
26+ pedestrians = pedestrian_classifier .detectMultiScale (gimg )
27+ cars = car_classifier .detectMultiScale (gimg )
28+ bike = bike_classifier .detectMultiScale (gimg )
29+ bus = bus_classifier .detectMultiScale (gimg )
30+
31+
32+ #Display the detection
33+ for (x ,y ,w ,h ) in cars :
34+ cv2 .rectangle (frame ,(x ,y ),(x + w ,y + h ),(0 ,255 ,0 ),4 )
35+ #Display the detection
36+ for (x ,y ,w ,h ) in bus :
37+ cv2 .rectangle (frame ,(x ,y ),(x + w ,y + h ),(0 ,0 ,225 ),4 )
38+ for (x ,y ,w ,h ) in bike :
39+ cv2 .rectangle (frame ,(x ,y ),(x + w ,y + h ),(255 ,0 ,0 ),4 )
40+ #Display the detection
41+ for (x ,y ,w ,h ) in pedestrians :
42+ cv2 .rectangle (frame ,(x ,y ),(x + w ,y + h ),(69 ,125 ,23 ),4 )
43+
44+ #display image
45+ cv2 .imshow ("Sasta Tesla Sofware XD" ,frame )
46+ cv2 .waitKey (1 )
47+ else :
48+ print ("Error in reading the frames" )
49+ break
50+ print ("Code Completed 👑" )
0 commit comments