feat(ai): 集成动漫人物分割与面部解析AI模型- 添加 DJL 深度学习框架依赖项以支持 PyTorch 和 ONNX Runtime 引擎

- 实现 Anime2VividModelWrapper 封装类用于动漫人物前景背景分离
- 开发 AnimeModelWrapper用于精细的动漫面部特征(如头发、眼睛)分割
- 创建配套的标签调色板和结果处理工具类提升可视化效果
- 增加多个测试用例验证不同AI模型的推理及文件输出功能
- 支持通过 synset.txt 自定义模型标签并增强命令行可测试性
This commit is contained in:
tzdwindows 7
2025-10-27 18:39:13 +08:00
parent f2cb74379e
commit a725e7eb23
18 changed files with 2414 additions and 0 deletions

View File

@@ -51,6 +51,16 @@ dependencies {
implementation files('libs/dog api 1.3.jar')
implementation files('libs/DesktopWallpaperSdk-1.0-SNAPSHOT.jar')
// === DJL API ===
implementation platform('ai.djl:bom:0.35.0')
implementation 'ai.djl:api'
implementation 'ai.djl:model-zoo'
implementation 'ai.djl.pytorch:pytorch-model-zoo:0.35.0'
implementation 'ai.djl.pytorch:pytorch-engine'
implementation 'ai.djl:basicdataset'
implementation 'ai.djl.onnxruntime:onnxruntime-engine'
runtimeOnly 'ai.djl.pytorch:pytorch-native-cpu:2.7.1'
runtimeOnly 'ai.djl.onnxruntime:onnxruntime-native-cpu:1.3.0'
// === 核心工具库 ===
implementation 'com.google.code.gson:gson:2.10.1' // 统一版本
implementation 'org.apache.logging.log4j:log4j-api:2.20.0'