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

最新推荐文章于 2022-07-11 14:13:59 发布

饿二饿二饿

最新推荐文章于 2022-07-11 14:13:59 发布

阅读量5.5k

点赞数 6

分类专栏： deep

本文链接：https://blog.csdn.net/eereere/article/details/80178595

版权

本文详细介绍了使用Tensorflow实现Mask-RCNN进行实时视频物体检测的步骤，包括模型训练、视频读取和帧处理，展示了如何在每一帧上应用预训练模型进行目标分割和定位。

摘要由CSDN通过智能技术生成

参考： https://www.youtube.com/watch?v=lLM8oAsi32g

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)]
    return colors

def apply_mask(image, mask, color, alpha=0.5):
    """Apply the given mask to the image.
    """
    for n, c in enumerate(color):
        image[:, :, n] = np.where(
            mask == 1,
            image[:, :, n] *(1 - alpha) + alpha * c,
            image[:, :, n]
        )
    return image

def display_instances(image,boxes,masks,ids,names,scores):
    n_instances=boxes.shape[0]
    if not n_instances:
        print('No instances to display')
    else:
        assert boxes.shape[0] == masks.shap