上面几篇文章已经初步讲解了
- 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即可