Python ResNet.residual_stage_detectron_mappingの例

コード例 #1

 def detectron_weight_mapping(self):
     detectron_weight_mapping, orphan_in_detectron = \
       ResNet.residual_stage_detectron_mapping(self.res5, 'res5', 3, 5)
     detectron_weight_mapping.update({
         'upconv5.weight': 'conv5_mask_w',
         'upconv5.bias': 'conv5_mask_b'
     })
     return detectron_weight_mapping, orphan_in_detectron

コード例 #2

ファイル: mask_rcnn_heads.py プロジェクト: chenyilun95/PANet

 def detectron_weight_mapping(self):
     detectron_weight_mapping, orphan_in_detectron = \
       ResNet.residual_stage_detectron_mapping(self.res5, 'res5', 3, 5)
     detectron_weight_mapping.update({
         'upconv5.weight': 'conv5_mask_w',
         'upconv5.bias': 'conv5_mask_b'
     })
     return detectron_weight_mapping, orphan_in_detectron

コード例 #3

    def detectron_weight_mapping(self):
        detectron_weight_mapping, orphan_in_detectron = \
          ResNet.residual_stage_detectron_mapping(self.res5, 'res5', 3, 5)
        # Assign None for res5 modules, do not load from or save to checkpoint
        for k in detectron_weight_mapping:
            detectron_weight_mapping[k] = None

        detectron_weight_mapping.update({
            'upconv5.weight': 'conv5_mask_w',
            'upconv5.bias': 'conv5_mask_b'
        })
        return detectron_weight_mapping, orphan_in_detectron

コード例 #4

ファイル: mask_rcnn_heads.py プロジェクト: chenyilun95/PANet

    def detectron_weight_mapping(self):
        detectron_weight_mapping, orphan_in_detectron = \
          ResNet.residual_stage_detectron_mapping(self.res5, 'res5', 3, 5)
        # Assign None for res5 modules, do not load from or save to checkpoint
        for k in detectron_weight_mapping:
            detectron_weight_mapping[k] = None

        detectron_weight_mapping.update({
            'upconv5.weight': 'conv5_mask_w',
            'upconv5.bias': 'conv5_mask_b'
        })
        return detectron_weight_mapping, orphan_in_detectron