类 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,则在 classify() 函数中的前向推断后应用 softmax,以将置信度范围转换为 [0.0-1.0]。 此函数允许您切换此行为。当模型中不包含 softmax 层时,请设置为 true。
        参数
        enable - 设置在 classify() 函数中启用 softmax 后处理。
        返回
        自动生成
      • getEnableSoftmaxPostProcessing

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

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

        protected void finalize()
                         throws java.lang.Throwable
        重写
        finalize 在类 Model
        抛出
        java.lang.Throwable