ncnn源码分析-002-layer

最新推荐文章于 2023-12-12 22:31:32 发布

原创最新推荐文章于 2023-12-12 22:31:32 发布 · 605 阅读

1 ·

CC 4.0 BY-SA版权

高性能计算专栏收录该内容

4 篇文章

订阅专栏

本文深入探讨了神经网络中的层结构，包括类结构定义、参数加载与模型加载过程。详细解析了load_param和load_model函数如何在layer及net类中实现参数解析，以及参数从文件读取到具体层成员变量的映射过程。

1.类结构

class Layer
{
public:
    int typeindex;
    std::string type;
    std::string name;
    std::vector<int> bottoms;//输入blob索引
    std::vector<int> tops;//输出blob索引

    int load_param(const ParamDict &pd);
    int load_model(const ModelBin &mb);
    int forward(const std::vector<Mat> &bottom_blobs, std::vector<Mat> &top_blobs, const Option &opt = get_default_option());
}

load_param和load_model两个函数除在layer中出现意外，还是net类结构里的成员函数，只是输入的参数类型不同，net类中通过FILE*加载，将参数转换为ParamDict结构，然后调用当前layer的成员函数int load_param(const ParamDict &pd)来解析param文件中的参数到具体的层结构成员变量中，实现参数的解析。