类 ClassificationModel


  • public class ClassificationModel
    extends Model
    此类表示分类模型的高级 API。ClassificationModel 允许设置预处理输入图像的参数。ClassificationModel 从包含训练权重和配置的文件中创建网络,设置预处理输入,运行前向传递并返回 top-1 预测。
    • 构造函数详情

      • ClassificationModel

        protected ClassificationModel​(long addr)
      • ClassificationModel

        public ClassificationModel​(java.lang.String model,
                                   java.lang.String config)
        从以一种受支持的格式表示的网络创建分类模型。modelconfig 参数的顺序无关紧要。
        参数
        model - 包含训练权重的二进制文件。
        config - 包含网络配置的文本文件。
      • ClassificationModel

        public ClassificationModel​(java.lang.String model)
        从以一种受支持的格式表示的网络创建分类模型。modelconfig 参数的顺序无关紧要。
        参数
        model - 包含训练权重的二进制文件。
      • ClassificationModel

        public ClassificationModel​(Net network)
        从深度学习网络创建模型。
        参数
        network - Net 对象。
    • 方法详情

      • setEnableSoftmaxPostProcessing

        public ClassificationModel setEnableSoftmaxPostProcessing​(boolean enable)
        设置启用/禁用 softmax 后处理选项。如果此选项为 true,则 softmax 将在 classify() 函数中的前向推理之后应用,以将置信度范围转换为 [0.0-1.0]。此函数允许您切换此行为。如果模型中不包含 softmax 层,请将其设置为 true。
        参数
        enable - 在 classify() 函数中设置启用 softmax 后处理。
        返回值
        自动生成
      • getEnableSoftmaxPostProcessing

        public boolean getEnableSoftmaxPostProcessing()
        获取启用/禁用 softmax 后处理选项。此选项默认为 false,在 classify() 函数中不应用 softmax 后处理。
        返回值
        自动生成
      • classify

        public void classify​(Mat frame,
                             int[] classId,
                             float[] conf)
      • finalize

        protected void finalize()
                         throws java.lang.Throwable
        覆盖
        finalize in class Model
        抛出
        java.lang.Throwable