def extract(cls, node):
proto_layer = node.pb
param = proto_layer.proposal_param
update_attrs = {
'feat_stride': param.feat_stride,
'base_size': param.base_size,
'min_size': param.min_size,
'ratio': np.array(param.ratio),
'scale': np.array(param.scale),
'pre_nms_topn': param.pre_nms_topn,
'post_nms_topn': param.post_nms_topn,
'nms_thresh': param.nms_thresh
}
mapping_rule = merge_attrs(param, update_attrs)
# update the attributes of the node
ProposalOp.update_node_stat(node, mapping_rule)
return cls.enabled
def extract_proposal_params(node, defaults):
param = node.pb.python_param
attrs = CaffePythonFrontExtractorOp.parse_param_str(param.param_str)
update_attrs = defaults
if 'ratios' in attrs and 'ratio' in attrs:
log.error(
'Both ratios and ratio found, value of ratios will be used',
extra={'is_warning': True})
if 'scales' in attrs and 'scale' in attrs:
log.error(
'Both scales and scale found, value of scales will be used',
extra={'is_warning': True})
if 'ratios' in attrs:
attrs['ratio'] = attrs['ratios']
del attrs['ratios']
if 'scales' in attrs:
attrs['scale'] = attrs['scales']
del attrs['scales']
update_attrs.update(attrs)
CaffePythonFrontExtractorOp.check_param(ProposalOp, update_attrs)
ProposalOp.update_node_stat(node, update_attrs)
def extract(self, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
feat_stride = attrs.int("feat_stride", 16)
ratio = attrs.tuple("ratios", float, (0.5, 1, 2))
scale = attrs.tuple("scales", int, (4, 8, 16, 32))
min_size = attrs.int("rpn_min_size", 16)
pre_nms_topn = attrs.int("rpn_pre_nms_top_n", 6000)
post_nms_topn = attrs.int("rpn_post_nms_top_n", 300)
nms_thresh = attrs.float("threshold", 0.7)
node_attrs = {
'feat_stride': feat_stride,
'base_size': 0,
'min_size': min_size,
'ratio': np.array(ratio),
'scale': np.array(scale),
'pre_nms_topn': pre_nms_topn,
'post_nms_topn': post_nms_topn,
'nms_thresh': nms_thresh,
}
ProposalOp.update_node_stat(node, node_attrs)
return (True, node_attrs)
def extract(cls, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
pre_nms_topn = attrs.int('rpn_pre_nms_top_n', 6000)
post_nms_topn = attrs.int('rpn_post_nms_top_n', 300)
nms_thresh = attrs.float('threshold', 0.7)
min_size = attrs.int('rpn_min_size', 16)
scale = attrs.tuple("scales", float, (4, 8, 16, 32))
ratio = attrs.tuple("ratios", float, (0.5, 1, 2))
feat_stride = attrs.int('feature_stride', 16)
update_attrs = {
'feat_stride': feat_stride,
'ratio': np.array(ratio),
'min_size': min_size,
'scale': np.array(scale),
'pre_nms_topn': pre_nms_topn,
'post_nms_topn': post_nms_topn,
'nms_thresh': nms_thresh,
'base_size': feat_stride
}
# update the attributes of the node
ProposalOp.update_node_stat(node, update_attrs)
return cls.enabled