Prathamesh1420 commited on
Commit
4374db8
·
verified ·
1 Parent(s): 811821c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -190,4 +190,20 @@ def main():
190
  cv2.putText(roi, str(i.getId()), (i.getX(), i.getY()), FONT, 0.3, TEXT_COLOR, 1, cv2.LINE_AA)
191
 
192
  str_cars = f'Cars: {cnt_cars}'
193
- str_trucks = f'T
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
190
  cv2.putText(roi, str(i.getId()), (i.getX(), i.getY()), FONT, 0.3, TEXT_COLOR, 1, cv2.LINE_AA)
191
 
192
  str_cars = f'Cars: {cnt_cars}'
193
+ str_trucks = f'Trucks: {cnt_trucks}'
194
+ cv2.putText(frame, str_cars, TEXT_POSITION_COUNT_CARS, FONT, TEXT_SIZE, TEXT_COLOR, 1, cv2.LINE_AA)
195
+ cv2.putText(frame, str_trucks, TEXT_POSITION_COUNT_TRUCKS, FONT, TEXT_SIZE, TEXT_COLOR, 1, cv2.LINE_AA)
196
+ cv2.line(frame, (w1, line_IN), (w1 + w2, line_IN), LINE_IN_COLOR, 1)
197
+ cv2.line(frame, (w1, line_OUT), (w1 + w2, line_OUT), LINE_OUT_COLOR, 1)
198
+
199
+ cv2.imshow("Vehicle Counting", roi)
200
+
201
+ k = cv2.waitKey(30) & 0xff
202
+ if k == 27:
203
+ break
204
+
205
+ cap.release()
206
+ cv2.destroyAllWindows()
207
+
208
+ if __name__ == "__main__":
209
+ main()