mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-14 22:18:42 +00:00
临时提交
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
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);
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user