georgewei / ai-service

基于djl实现的深度学习服务

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

基于djl实现的深度学习服务


Languages

Language:Java 100.0%