cn.smartjavaai.vision.utils

类 CategoryMaskFilter

java.lang.Object

cn.smartjavaai.vision.utils.CategoryMaskFilter

public class CategoryMaskFilter
extends Object

CategoryMask过滤器

作者:

dwj

构造器概要

构造器

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

方法概要

限定符和类型	方法和说明
ai.djl.modality.cv.output.CategoryMask	filter(ai.djl.modality.cv.output.CategoryMask categoryMask) 对给定的CategoryMask进行过滤，只保留允许的分类。

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

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

构造器详细资料

CategoryMaskFilter

public CategoryMaskFilter(List<String> allowedClasses)

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

参数:

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

方法详细资料

filter

public ai.djl.modality.cv.output.CategoryMask filter(ai.djl.modality.cv.output.CategoryMask categoryMask)

对给定的CategoryMask进行过滤，只保留允许的分类。

参数:

categoryMask - 待过滤的CategoryMask

返回:

过滤后的CategoryMask