TensorFlow Object Detection API视频检测

上面几篇文章已经初步讲解了

  • TensorFlow Object Detection API的安装使用
  • 利用TensorFlow Object Detection API进行小浣熊的检测(图片)

这节讲解利用 TensorFlow Object Detection API (基于已经训练好的小浣熊检测模型)进行视频检测~

这是参考博客:https://blog.csdn.net/xiaoxiao123jun/article/details/76605928

一 安装几个包

首先我们需要安装opencv-python

命令
pip install opencv-python
安装完成,试下能不能导进去,没报错就ok了
>>>>import cv2
>>>>

如果再提示No moudle name xxx 的话,你就pip install xxx 就可以了

二 源码

源码的地址是:https://github.com/priya-dwivedi/Deep-Learning/blob/master/Object_Detection_Tensorflow_API.ipynb

我们复制一下:
代码大家到源码里面扣
这里的视频检测接在小浣熊检测代码的最后面~
对了,还要import cv2 忘了贴了

如果报错视频assignment destination is read-only
要改动image_process = detect_objects(np.array(image), sess, detection_graph)
然后重启notebook即可
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/m0_37970224/article/details/88837494