|
| Layer () |
|
| Layer (const LayerParams ¶ms) |
| 仅初始化 name、type 和 blobs 字段。
|
|
virtual | ~Layer () |
|
virtual void | applyHalideScheduler (Ptr< BackendNode > &node, const std::vector< Mat * > &inputs, const std::vector< Mat > &outputs, int targetId) const |
| 基于层超参数自动 Halide 调度。
|
|
virtual void | finalize (const std::vector< Mat * > &element, std::vector< Mat > &output) |
| 根据输入、输出和数据块计算并设置内部参数。
|
|
std::vector< Mat > | finalize (const std::vector< Mat > &element) |
| 这是重载的成员函数,供便利使用。它仅在接受的参数上与上述函数不同。
|
|
void | finalize (const std::vector<Mat > &inputs, const std::vector<Mat > &outputs) |
| 这是重载的成员函数,供便利使用。它仅在接受的参数上与上述函数不同。
|
|
virtual void | InputArrayOfArrays inputs, | | 根据输入、输出和数据块计算并设置内部参数。
|
|
virtual void | forward ( | | 给定inputs 张量,计算输出blobs 。
|
|
virtual void | forward (std::vector< Mat*> &inputs, std::vector<<(a class="el" href))/(classcv_1_1Mat.html)> Mat> &outputs, std::vector<<(a class="el" href))/(classcv_1_1Mat.html)> Mat> &internals) |
| 给定inputs 张量,计算输出blobs 。
|
|
void | forward_fallback ( | | 给定inputs 张量,计算输出blobs 。
|
|
虚拟 getFLOPS (const std::vector<<(a class="el" href))/(group__dnn.html#ga8a9ab61770c140f0fa2880c90aeae832)> MatShape > &inputs, const std::vector<<(a class="el" href))/(group__dnn.html#ga8a9ab61770c140f0fa2880c90aeae832)> MatShape > &outputs) const |
|
虚拟bool& | getMemoryShapes (const std::vector<<(a class="el" href))/(group__dnn.html#ga8a9ab61770c140f0fa2880c90aeae832)> MatShape > &inputs, const int requiredOutputs, std::vector< MatShape > &outputs, std::vector< MatShape > &internals) const |
|
virtual void | getScaleShift ( Mat &scale, Mat &shift) const |
| 返回具有通道乘法和加法的层的参数。
|
|
virtual void | getScaleZeropoint (float &scale, int &zeropoint) const |
| 返回层的缩放和偏移量。
|
|
虚拟 Ptr>< BackendNode> | initCann (const std::vector< Ptr>< BackendWrapper> &inputs, const std::vector< Ptr>< BackendWrapper> &outputs, const std::vector< Ptr>< BackendNode> &nodes) |
| 返回一个CANN后端节点。
|
|
虚拟 Ptr>< BackendNode> | initCUDA (void *context, const std::vector< Ptr>< BackendWrapper> &inputs, const std::vector< Ptr>< BackendWrapper> &outputs) |
| 返回一个CUDA后端节点。
|
|
虚拟 Ptr>< BackendNode> | initHalide (const std::vector< PtrBackendWrapper > > &inputs) |
| 返回Halide后端节点。
|
|
虚拟 Ptr>< BackendNode> | initNgraph (const std::vector< PtrBackendWrapper > > &inputs, const std::vector< PtrBackendNode > > &nodes) |
|
虚拟 Ptr>< BackendNode> | initTimVX (void *timVxInfo, const std::vector< PtrBackendWrapper > > &inputsWrapper, const std::vector< PtrBackendWrapper > > &outputsWrapper, bool isLast) |
| 返回一个TimVX后端节点。
|
|
虚拟 Ptr>< BackendNode> | initVkCom (const std::vector< PtrBackendWrapper > > &inputs, std::vector< PtrBackendWrapper > > &outputs) |
|
虚拟 Ptr>< BackendNode> | initWebnn (const std::vector< PtrBackendWrapper > > &inputs, const std::vector< PtrBackendNode > > &nodes) |
|
virtual int | inputNameToIndex (String inputName) |
| 返回输入blob在输入数组中的索引。
|
|
virtual int | outputNameToIndex (const String &outputName) |
| 返回输出blob在输出数组中的索引。
|
|
void | run (const std::vector< Mat > &inputs, std::vector< Mat > &outputs, std::vector< Mat > &internals) |
| 分配图层并计算输出。
|
|
虚拟bool& | setActivation (const PtrActivationLayer > &layer) |
| 尝试将后续的激活层附加到图层上,即在一部分情况的层融合中。
|
|
void | setParamsFrom (const LayerParams ¶ms) |
| 仅初始化 name、type 和 blobs 字段。
|
|
虚拟bool& | supportBackend (int backendId) |
| 询问图层是否支持特定的后端进行计算。
|
|
虚拟 Ptr>< BackendNode> | tryAttach (const PtrBackendNode > &node) |
| 实现层融合。
|
|
虚拟bool& | tryFuse (PtrLayer > &top) |
| 尝试将当前图层与下一个图层融合。
|
|
虚拟bool& | tryQuantize (const std::vector< std::vector< float > > &scales, const std::vector< std::vector< int > > &zeropoints, LayerParams ¶ms) |
| 尝试量化给定的图层,并计算固定点实现所需的量化参数。
|
|
virtual void | 取消附加 () |
| “解除”特定层上所有层的连接。
|
|
虚拟bool& | updateMemoryShapes (const std::vector< MatShape > & inputs) |
|
| Algorithm () |
|
virtual | ~Algorithm () |
|
virtual void | clear () |
| 清除算法状态。
|
|
虚拟bool& | empty () const |
| 如果 Algorithm 是空的(例如在最开始或读取失败后),则返回 true。
|
|
virtual < String > | getDefaultName () const |
|
virtual void | read (const FileNode & fn) |
| 从文件存储中读取算法参数。
|
|
virtual void | save (const String & filename) const |
|
void | write (const Ptr< FileStorage > & fs, const String & name=String()) const |
|
virtual void | write (FileStorage & fs) const |
| 将算法参数存储在文件存储中。
|
|
void | write (FileStorage & fs, const String & name) const |
|
std::vector< Mat > | blobs |
| 必须在此处存储学习到的参数列表,以便使用 Net::getParam() 读取它们。
|
|
< String > | name |
| 层的实例名称,可用于日志记录或其他内部用途。
|
|
int | preferableTarget |
| 偏好层前向传递的目标
|
|
< String > | type |
| 创建层时使用的类型名称,由层工厂使用。
|
|
void | writeFormat (FileStorage & fs) const |
|