diff --git a/vision/src/main/java/cn/smartjavaai/objectdetection/criteria/CriteriaBuilderFactory.java b/vision/src/main/java/cn/smartjavaai/objectdetection/criteria/CriteriaBuilderFactory.java index 10a70ab..753b402 100644 --- a/vision/src/main/java/cn/smartjavaai/objectdetection/criteria/CriteriaBuilderFactory.java +++ b/vision/src/main/java/cn/smartjavaai/objectdetection/criteria/CriteriaBuilderFactory.java @@ -29,10 +29,14 @@ public class CriteriaBuilderFactory { switch (config.getModelEnum()) { case YOLOV8_OFFICIAL_ONNX: return new YoloCriteriaBuilder().buildCriteria(config); + case YOLOV11_OFFICIAL_ONNX: + return new YoloCriteriaBuilder().buildCriteria(config); case YOLOV12_OFFICIAL_ONNX: return new YoloCriteriaBuilder().buildCriteria(config); case YOLOV8_CUSTOM_ONNX: return new YoloCriteriaBuilder().buildCriteria(config); + case YOLOV11_CUSTOM_ONNX: + return new YoloCriteriaBuilder().buildCriteria(config); case YOLOV12_CUSTOM_ONNX: return new YoloCriteriaBuilder().buildCriteria(config); case TENSORFLOW2_OFFICIAL: