easy_vision.python.model.mask_rcnn

easy_vision.python.model.mask_rcnn.mrcnn_head

class easy_vision.python.model.mask_rcnn.mrcnn_head.MRCNNHead(feature_dict, head_config, label_dict=None, fpn_config=None, mode='predict', region_feature_extractor=None)

Bases: easy_vision.python.model.cv_head.CVHead

for the third stage of faster rcnn: instance mask prediction

__init__(feature_dict, head_config, label_dict=None, fpn_config=None, mode='predict', region_feature_extractor=None)

Parameters:

• feature_dict – input dict of features
• head_config – rcnn head config
• label_dict – a dict of labels, during prediction, it can be None
• fpn_config – config of fpn
• mode – train for train phase, evaluate for evaluate phase, predict for predict phase
• region_feature_extractor – block reuse part of backbone to extract box feature in second stage

build_loss_graph()

build_postprocess_graph()

for post process of the head it is necessary to separate predict and postprocess because postprocess is not needed during train but needed during test

build_predict_graph()

Predicts non-box, non-class outputs using refined detections.