Tensorflow Mask-RCNN（二）——非实时 检测视频

代码参考：https://github.com/Tony607/colab-mask-rcnn

具体安装请见上一篇博客

分两步走：

①把下载好的视频变成一帧一帧的，对每一帧进行detection，把框，label，scores都标在图上，保存成图片

② 把保存好的一帧一帧的图片，合成视频

有些同学可能不太懂怎么获取摄像头

下面的代码是一个小demo——如何获取摄像头或者获取本地视频，并显示在当前窗口上

import cv2
import numpy as np#添加模块和矩阵模块
cap=cv2.VideoCapture(0)
#打开摄像头，若打开本地视频，同opencv一样，只需将０换成("×××.avi")
while(1):    # get a frame   
    ret, frame = cap.read()    # show a frame   
    cv2.imshow("capture", frame)   
    if cv2.waitKey(1) & 0xFF == ord('q'):        
        break
cap.release()
cv2.destroyAllWindows()

第一步： 把下载好的视频变成一帧一帧的，用Mask RCNN检测物体

import cv2
import numpy as np
# 定义随机颜色函数
def random_colors(N):
    np.random.seed(1)
    colors=[tuple(255 * np.random.rand(3)) for _ in range(N)