cn.smartjavaai.vision.utils

类 ClassificationFilter

java.lang.Object

cn.smartjavaai.vision.utils.ClassificationFilter

public class ClassificationFilter
extends Object

Classification过滤器

作者:

dwj

构造器概要

构造器

构造器和说明
ClassificationFilter(List<String> allowedClasses, float threshold) 构造函数，初始化允许的分类列表和置信度阈值

方法概要

限定符和类型	方法和说明
ai.djl.modality.Classifications	filter(ai.djl.modality.Classifications classifications) 过滤分类结果并返回新的Classifications对象

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

ClassificationFilter

public ClassificationFilter(List<String> allowedClasses,
                            float threshold)

构造函数，初始化允许的分类列表和置信度阈值

参数:

allowedClasses - 允许的分类列表，null表示不过滤分类

threshold - 置信度阈值

方法详细资料

filter

public ai.djl.modality.Classifications filter(ai.djl.modality.Classifications classifications)

过滤分类结果并返回新的Classifications对象

参数:

classifications - 原始分类结果

返回:

过滤后的Classifications对象