代码拉取完成,页面将自动刷新
/**
* 初始文本检测推理器
*/
public static Predictor<Image, DetectedObjects> loadPredictor() throws ModelNotFoundException, MalformedModelException, IOException {
try (ZooModel<Image, DetectedObjects> detectionModel = ModelZoo.loadModel(getCriteria())) {
return detectionModel.newPredictor();
} catch (Exception e) {
System.out.println("初始文本检测推理器失败!");
throw new RuntimeException(e);
}
}
private static Criteria<Image, DetectedObjects> getCriteria() {
return Criteria.builder()
.optEngine("OnnxRuntime")
.optModelName("inference")
.setTypes(Image.class, DetectedObjects.class)
.optModelPath(Paths.get("src/main/resources/text/models/ch_PP-OCRv3_det_infer.tar"))
.optTranslator(new PpWordDetectionTranslator(new ConcurrentHashMap<String, String>()))
.optProgress(new ProgressBar())
.build();
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。
我这样引入会报错,提示ai.djl.repository.zoo.ModelNotFoundException: No model with the specified URI or the matching Input/Output type is found.
我怀疑是setTypes给的值不对,但我不知道正确的值是什么
清楚了,不是.optEngine("OnnxRuntime")是.optEngine("PaddlePaddle")
登录 后才可以发表评论