使用到的库

好多人都想了解一下如何对摄像头进行调用，然后进行目标检测，于是我做了这个小BLOG。

opencv-python==4.1.2.30

Pillow==6.2.1

numpy==1.17.4

这些都是通用的库，版本不同问题应该也不大。

实现思路

利用opencv调用摄像头，读取每一帧传入目标检测网络检测，将检测结果呈现。

由于本文所用的检测格式为RGB格式，CV2读取的时候会使用BGR格式，因此在检测的时候要利用cv2.cvtColor进行转换。

实现代码

以Retinanet为例：

from keras.layers import Input
from retinanet import Retinanet
from PIL import Image
import numpy as np
import cv2
retinanet = Retinanet()
# 调用摄像头
capture=cv2.VideoCapture(0) 
while(True):
    # 读取某一帧
    ref,frame=capture.read()
    # 格式转变，BGRtoRGB
    frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
    # 转变成Image
    frame = Image.fromarray(np.uint8(frame))
    # 进行检测
    frame = np.array(retinanet.detect_image(frame))
    # RGBtoBGR满足opencv显示格式
    frame = cv2.cvtColor(frame,cv2.COLOR_RGB2BGR)
    cv2.imshow("video",frame)
    c= cv2.waitKey(30) & 0xff 
    if c==27:
        capture.release()
        break
retinanet.close_session()

2020/4/26更新：FPS计算

很多小伙伴说到想要获取FPS，于是给所有的目标检测网络加上了FPS的功能，需要的小伙伴请重新下载。

FPS记录的原理

FPS简单来理解就是图像的刷新频率，也就是每秒多少帧。

假设目标检测网络处理1帧要0.02s。

此时FPS就是50。

FPS实现代码

该代码以ssd为例。

#-------------------------------------#
#       调用摄像头检测
#-------------------------------------#
from ssd import SSD
from PIL import Image
import numpy as np
import cv2
import time
ssd = SSD()
# 调用摄像头
capture=cv2.VideoCapture(0) # capture=cv2.VideoCapture("1.mp4")
fps = 0.0
while(True):
    t1 = time.time()
    # 读取某一帧
    ref,frame=capture.read()
    # 格式转变，BGRtoRGB
    frame = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
    # 转变成Image
    frame = Image.fromarray(np.uint8(frame))
    # 进行检测
    frame = np.array(ssd.detect_image(frame))
    # RGBtoBGR满足opencv显示格式
    frame = cv2.cvtColor(frame,cv2.COLOR_RGB2BGR)
    fps  = ( fps + (1./(time.time()-t1)) ) / 2
    print("fps= %.2f"%(fps))
    frame = cv2.putText(frame, "fps= %.2f"%(fps), (0, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
    cv2.imshow("video",frame)
    c= cv2.waitKey(30) & 0xff 
    if c==27:
        capture.release()
        break

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

猜你喜欢

• c++中怎么样调用Python中封装的函数或接口？

  本文给大家介绍的是c++ 调用python的内容，下文是c++中怎么样调用Python中封装的函数或接口的示例，有很详细的介绍，有这方面学习需要的朋友可以了解看看，那么接下来就跟随小编一起来学习一下吧。

• opencv模板匹配中边框重复的情况怎么解决

  这篇文章主要介绍了opencv模板匹配相同位置去除重复的框，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

• Python命令行库click能做什么，用法是怎样的

  本文主要介绍了Python命令行库click的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

• python常用的内置对象有什么？数据类型你了解多少？

  这篇文章主要介绍Python内置对象的内容，在Python学习中，python常用的内置对象是比较基础的知识，下文给大家简单介绍常用的python内置对象、变量、整数、浮点数、布尔类型、字符串类型和注释等知识，有需要的朋友可以看看，希望大家阅读完这篇文章能有所收获。

• 浅谈python装饰器的原理、作用和使用

  今天我们来总结一下python装饰器的原理、作用和使用，之前我们以及有详细的了解过python装饰器了，因此这篇文章就带大家简单的回顾一下python装饰器的知识，有需要的朋友可以参考。