Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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()
|