mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-12 20:58:51 +00:00
23 lines
617 B
Java
23 lines
617 B
Java
package cn.smartjavaai.objectdetection.criteria;
|
|
|
|
import ai.djl.modality.cv.Image;
|
|
import ai.djl.modality.cv.output.DetectedObjects;
|
|
import ai.djl.repository.zoo.Criteria;
|
|
import cn.smartjavaai.objectdetection.config.DetectorModelConfig;
|
|
|
|
/**
|
|
* 模型加载策略接口，用于根据不同模型类型构建对应的 DJL Criteria 实例
|
|
* @author dwj
|
|
* @date 2025/5/14
|
|
*/
|
|
public interface CriteriaBuilderStrategy {
|
|
|
|
/**
|
|
* 根据模型类型构建对应的 DJL Criteria 实例
|
|
* @param config
|
|
* @return
|
|
*/
|
|
Criteria<Image, DetectedObjects> buildCriteria(DetectorModelConfig config);
|
|
|
|
}
|