10行Python代码计算汽车数量的实现方法

当你还是个孩子坐车旅行的时候,你玩过数经过的汽车的数目的游戏吗?

在这篇文章中,我将教你如何使用10行Python代码构建自己的汽车计数程序。以下是环境及相应的版本库:

  • Python版本 3.6.9
  • cvlib: 0.2.2
  • opencv-python: 4.1.1.26
  • tensorflow: 1.14.0
  • matplotlib: 3.1.1
  • Keras: 2.2.5

下面的代码用于导入所需的python库、从存储中读取图像、对图像执行目标检测、用边界框显示图像以及关于检测目标的标签、计算图像中的汽车数量并打印图像:

import cv2
import matplotlib.pyplot as plt
import cvlib as cv
from cvlib.object_detection import draw_bbox
im = cv2.imread('cars_4.jpeg')
bbox, label, conf = cv.detect_common_objects(im)
output_image = draw_bbox(im, bbox, label, conf)
plt.imshow(output_image)
plt.show()
print('Number of cars in the image is ' str(label.count('car')))

此图像的输出:

Number of cars in the image is 22

此图像的输出:

Number of cars in the image is 25

你的汽车计数器程序现在已经准备好了。你可以用它来做一些有趣的实验,比如计算每天经过你家门前车道的汽车数量。

要了解更多关于cvlib库的信息,可以访问下面的链接。

cvlib: https://www.cvlib.net

欢迎关注PyTorch官方中文教程站:http://pytorch.panchuang.net/

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持来客网。