基于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+
- brain提供的自动抠图AI初始化
- 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接着启动SpringBoot应用
浏览器打开: http://localhost:8080/
实际体验效果如下：