Springboot+DJL集成深度学习实现OCR和对象检测服务

0x0 简介

DJL全称deep java library,官网https://djl.ai,是一个可以让java程序员快速集成深度学习的框架,你无需对神经网络算法有多么深入的了解就可以快速搭建一个ai服务器。

本项目基于DJL+springboot开发,OCR推理引擎使用paddle+pytorch,对象检测推理引擎使用onnx+pytorch;
OCR深度学习模型采用百度paddle的ocr模型,支持快速识别和精确识别两种类型;对象检测模型使用yolov5。

项目已发布至github:https://github.com/gx304419380/ai-service
注意:若使用gpu运算,你的电脑需要安装cuda环境,具体方法请自行百度

0x1 项目介绍

项目共有两个模块:ocr和yolo,其中ocr用于文字识别(Optical Character Recognition),yolo模块用于图片对象检测,具体如下图所示:


image.png

项目启动后可以进入swagger进行接口使用 http://localhost:8080/swagger-ui/

image.png

ocr相关功能测试图如下:


ocr.png

yolo对象检测相关功能测试图如下:


微信截图_20220319151847.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容