/autoRMBG

auto remove image backgound service by SpringBoot3+ JDK17+ AI

Primary LanguagePython

autoRMBG

基于SpringBoot3.2.2 + jdk17 + Thymeleaf + WebClient + briaai/RMBG-1.4 实现的基于AI的自动抠图服务

系列教程

启动方式

技术栈

自动抠图服务，目前主要依赖两套技术栈，分别是基于python的AI模型执行过程与基于Java的WEB应用

JAVA后端技术栈: JDK17 + Maven + SpringBoot3.2.2

基础组件：

技术 说明 官网
Spring & SpringMVC & WebFlux Java全栈应用程序框架和WEB容器实现 https://spring.io/
SpringBoot Spring应用集成开发框架 https://spring.io/projects/spring-boot
https 证书 https://letsencrypt.org/
lombok Java语言增强库 https://projectlombok.org
guava google开源的java工具集 https://github.com/google/guava
thymeleaf html5模板引擎 https://www.thymeleaf.org
jackson json/xml处理 https://www.jackson.com
bootstrap-fileinput 图片上传控件 https://github.com/kartik-v/bootstrap-fileinput
base-plugin 文件处理 https://github.com/liuyueyi/quick-media

Python技术栈: python3.11 + torch + fastapi

基础组件:

技术 说明 官网
torch 深度学习框架 PyTorch
Pillow 图像操作 pillow · PyPI
NumPy 科学计算基础包 NumPy -
huggingface_hub huggingface模型包管理工具 Hugging Face – The AI community building the future.
fastAPI web框架 FastAPI

环境依赖

  • python3.11
  • jdk17
  • maven3.6+

环境配置

  1. brain提供的自动抠图AI初始化

rmbg-ai服务安装

  1. jdk相关环境初始化
  • 安装jdk，maven
  • idea 加载项目，编译

启动服务

首先进入 rmbg-ai 启动服务

# mac/linux 环境
rmbg-ai/venv/bin/python rmbg-ai/main.py

# win 环境
.\rmbg-ai\venv\Scripts\python.exe .\rmbg-ai\main.py

ai启动

接着启动SpringBoot应用

Spring应用启动

浏览器打开: http://localhost:8080/

实际体验效果如下：