码迷,mamicode.com
首页 > 其他好文 > 详细

『计算机视觉』Mask-RCNN_其七:Mask生成(待续)

时间:2018-10-23 18:03:37      阅读:531      评论:0      收藏:0      [点我收藏+]

标签:alt   for   目标   input   png   生成   分享   pytho   layer   

一、Mask生成概览

技术分享图片

 

上一节的末尾,我们已经获取了待检测图片的分类回归信息,我们将回归信息(即待检测目标的边框信息)单独提取出来,结合金字塔特征mrcnn_feature_maps,进行Mask生成工作。

            # Detections
            # output is [batch, num_detections, (y1, x1, y2, x2, class_id, score)] in
            # normalized coordinates
            detections = DetectionLayer(config, name="mrcnn_detection")(
                [rpn_rois, mrcnn_class, mrcnn_bbox, input_image_meta])

            # Create masks for detections
            detection_boxes = KL.Lambda(lambda x: x[..., :4])(detections)
            mrcnn_mask = build_fpn_mask_graph(detection_boxes, mrcnn_feature_maps,
                                              input_image_meta,
                                              config.MASK_POOL_SIZE,
                                              config.NUM_CLASSES,
                                              train_bn=config.TRAIN_BN)

 

『计算机视觉』Mask-RCNN_其七:Mask生成(待续)

标签:alt   for   目标   input   png   生成   分享   pytho   layer   

原文地址:https://www.cnblogs.com/hellcat/p/9837595.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!