Function LoadCaffeRPN

tengine-module/plugin/serializer/caffe/caffe_serializer.cpp:994–1023

    return true;
}
static bool LoadCaffeRPN(StaticGraph* graph, StaticNode* node, const te_caffe::LayerParameter& layer_param)
{
    RPNParam param = any_cast<RPNParam>(OpManager::GetOpDefParam("RPN"));

    const te_caffe::RPNParameter& caffe_param = layer_param.rpn_param();
    //
    param.feat_stride = caffe_param.feat_stride();
    param.basesize = caffe_param.basesize();
    param.min_size = caffe_param.boxminsize();
    param.per_nms_topn = caffe_param.per_nms_topn();
    param.post_nms_topn = caffe_param.post_nms_topn();
    param.nms_thresh = caffe_param.nms_thresh();

    for(int i = 0; i < caffe_param.scale_size(); ++i)
    {
        param.anchor_scales.push_back(caffe_param.scale(i));
    }
    for(int i = 0; i < caffe_param.ratio_size(); ++i)
    {
        param.ratios.push_back(caffe_param.ratio(i));
    }

    StaticOp* op = CreateStaticOp(graph, "RPN");

    SetOperatorParam(op, param);
    SetOperatorDynamicShape(op);
    SetNodeOp(node, op);

    return true;
}
static bool LoadCaffeDetectionOutput(StaticGraph* graph, StaticNode* node, const te_caffe::LayerParameter& layer_param)
{
    DetectionOutputParam param = any_cast<DetectionOutputParam>(OpManager::GetOpDefParam("DetectionOutput"));